From 2ac38adbe5f4e1374a079e032ed4b504351a207c Mon Sep 17 00:00:00 2001 From: zhifu gao <zhifu.gzf@alibaba-inc.com> Date: 星期二, 23 四月 2024 18:08:57 +0800 Subject: [PATCH] Dev gzf exp (#1647) --- runtime/python/onnxruntime/funasr_onnx/vad_bin.py | 10 ++++++---- 1 files changed, 6 insertions(+), 4 deletions(-) diff --git a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py index 6b3a1bc..a2f443a 100644 --- a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py @@ -31,7 +31,8 @@ quantize: bool = False, intra_op_num_threads: int = 4, max_end_sil: int = None, - cache_dir: str = None + cache_dir: str = None, + **kwargs ): if not Path(model_dir).exists(): @@ -62,7 +63,7 @@ "\npip3 install -U funasr -i https://mirror.sjtu.edu.cn/pypi/web/simple" model = AutoModel(model=model_dir) - model_dir = model.export(type="onnx", quantize=quantize) + model_dir = model.export(type="onnx", quantize=quantize, **kwargs) config_file = os.path.join(model_dir, 'config.yaml') cmvn_file = os.path.join(model_dir, 'am.mvn') config = read_yaml(config_file) @@ -196,7 +197,8 @@ quantize: bool = False, intra_op_num_threads: int = 4, max_end_sil: int = None, - cache_dir: str = None + cache_dir: str = None, + **kwargs ): if not Path(model_dir).exists(): try: @@ -226,7 +228,7 @@ "\npip3 install -U funasr -i https://mirror.sjtu.edu.cn/pypi/web/simple" model = AutoModel(model=model_dir) - model_dir = model.export(type="onnx", quantize=quantize) + model_dir = model.export(type="onnx", quantize=quantize, **kwargs) config_file = os.path.join(model_dir, 'config.yaml') cmvn_file = os.path.join(model_dir, 'am.mvn') -- Gitblit v1.9.1