From a7d7a0f3a2e7cd44a337ced34e3536b12ccb534e Mon Sep 17 00:00:00 2001 From: zhifu gao <zhifu.gzf@alibaba-inc.com> Date: 星期一, 11 三月 2024 19:24:44 +0800 Subject: [PATCH] Dev gzf (#1467) --- runtime/python/onnxruntime/funasr_onnx/vad_bin.py | 8 ++++---- 1 files changed, 4 insertions(+), 4 deletions(-) diff --git a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py index af32b1d..384f377 100644 --- a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py @@ -61,8 +61,8 @@ "For the users in China, you could install with the command:\n" \ "\npip3 install -U funasr -i https://mirror.sjtu.edu.cn/pypi/web/simple" - model = AutoModel(model=cache_dir) - model_dir = model.export(type="onnx", quantize=quantize, device="cpu") + model = AutoModel(model=model_dir) + model_dir = model.export(type="onnx", quantize=quantize) config_file = os.path.join(model_dir, 'vad.yaml') cmvn_file = os.path.join(model_dir, 'vad.mvn') config = read_yaml(config_file) @@ -225,8 +225,8 @@ "For the users in China, you could install with the command:\n" \ "\npip3 install -U funasr -i https://mirror.sjtu.edu.cn/pypi/web/simple" - model = AutoModel(model=cache_dir) - model_dir = model.export(type="onnx", quantize=quantize, device="cpu") + model = AutoModel(model=model_dir) + model_dir = model.export(type="onnx", quantize=quantize) config_file = os.path.join(model_dir, 'vad.yaml') cmvn_file = os.path.join(model_dir, 'vad.mvn') -- Gitblit v1.9.1