From 7a9c0414b685b6f2ad7538017e2a1215e1e76901 Mon Sep 17 00:00:00 2001 From: Shi Xian <40013335+R1ckShi@users.noreply.github.com> Date: 星期二, 11 六月 2024 11:42:15 +0800 Subject: [PATCH] Merge pull request #1791 from dtlzhuangz/zhuangzhong_dev --- examples/industrial_data_pretraining/bicif_paraformer/export.py | 12 +++++++++--- 1 files changed, 9 insertions(+), 3 deletions(-) diff --git a/examples/industrial_data_pretraining/bicif_paraformer/export.py b/examples/industrial_data_pretraining/bicif_paraformer/export.py index d1b1ea7..31098d2 100644 --- a/examples/industrial_data_pretraining/bicif_paraformer/export.py +++ b/examples/industrial_data_pretraining/bicif_paraformer/export.py @@ -7,7 +7,10 @@ from funasr import AutoModel -model = AutoModel(model="iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch", device="cpu") +model = AutoModel( + model="iic/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-pytorch", + device="cpu", +) res = model.export(type="onnx", quantize=False) print(res) @@ -16,7 +19,10 @@ # method2, inference from local path from funasr import AutoModel -model = AutoModel(model="/Users/zhifu/.cache/modelscope/hub/iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch", device="cpu") +model = AutoModel( + model="/Users/zhifu/.cache/modelscope/hub/iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch", + device="cpu", +) res = model.export(type="onnx", quantize=False) -print(res) \ No newline at end of file +print(res) -- Gitblit v1.9.1