From 35b1c051f6db3649a818547902497d219c871b84 Mon Sep 17 00:00:00 2001 From: zhifu gao <zhifu.gzf@alibaba-inc.com> Date: 星期四, 14 三月 2024 09:33:30 +0800 Subject: [PATCH] Dev gzf llm (#1493) --- 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