zhifu gao
2024-03-14 35b1c051f6db3649a818547902497d219c871b84
funasr/utils/export_utils.py
@@ -4,8 +4,6 @@
def export_onnx(model,
                data_in=None,
            quantize: bool = False,
            fallback_num: int = 5,
            calib_num: int = 100,
            opset_version: int = 14,
            **kwargs):
   model_scripts = model.export(**kwargs)
@@ -19,8 +17,6 @@
      _onnx(m,
            data_in=data_in,
            quantize=quantize,
            fallback_num=fallback_num,
            calib_num=calib_num,
            opset_version=opset_version,
            export_dir=export_dir,
            **kwargs