funasr/utils/export_utils.py @@ -22,6 +22,7 @@ ) elif type == 'torchscripts': device = 'cuda' if torch.cuda.is_available() else 'cpu' print("Exporting torchscripts on device {}".format(device)) _torchscripts( m, path=export_dir,