funasr/utils/export_utils.py
@@ -3,10 +3,7 @@ def export_onnx(model, data_in=None, type: str = "onnx", quantize: bool = False, fallback_num: int = 5, calib_num: int = 100, opset_version: int = 14, **kwargs): model_scripts = model.export(**kwargs) @@ -19,10 +16,7 @@ m.eval() _onnx(m, data_in=data_in, type=type, quantize=quantize, fallback_num=fallback_num, calib_num=calib_num, opset_version=opset_version, export_dir=export_dir, **kwargs
