R1ckShi
2024-06-11 0f1247d7a8d481a7b5825c903e819dc543224740
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,