zhifu gao
2024-06-20 e65b1f701abca03bf3a1b5fbb200392aabd38c22
funasr/utils/export_utils.py
@@ -5,7 +5,7 @@
try:
    import torch_blade
except Exception as e:
    print(f"failed to load torch_blade: {e}")
    print(f"Warning, if you are exporting bladedisc, please install it and try it again: pip install -U torch_blade\n")
def export(model, data_in=None, quantize: bool = False, opset_version: int = 14, type='onnx', **kwargs):