modelscope / FunASR

A Fundamental End-to-End Speech Recognition Toolkit and Open Source SOTA Pretrained Models, Supporting Speech Recognition, Voice Activity Detection, Text Post-processing etc.
https://www.funasr.com
Other
6.5k stars 689 forks source link

Paraformer finetuned 热词模型onnx无法导出model_eb_quant.onnx #1057

Closed ben-8878 closed 11 months ago

ben-8878 commented 11 months ago

OS: centos7.9 Python/C++ Version:python3.8 gcc8.2 Package Version:funasr-onnx==0.2.3 pytorch==1.13.1、modelscope==1.6.1、funasr==0.5.8、torchaudio==0.13.1 Model:speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404-finetune Command:python -m funasr.export.export_model --model-name speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404-finetune --export-dir output --quantize true Details: 使用以上命令只能到处model.onnx和model_quant.onnx,无法导出model_eb_quant.onnx;使用https://github.com/alibaba-damo-academy/FunASR/blob/main/funasr/runtime/python/onnxruntime/demo_contextual_paraformer.py脚本测试,报以下错误。 Error log: File "/home/ybZhang/miniconda3/envs/modelscope/lib/python3.8/site-packages/funasr_onnx/paraformer_bin.py", line 286, in init File "/home/ybZhang/miniconda3/envs/modelscope/lib/python3.8/site-packages/funasr_onnx/paraformer_bin.py", line 286, in init self.ort_infer_eb = OrtInferSession(model_eb_file, device_id, intra_op_num_threads=intra_op_num_threads) File "/home/ybZhang/miniconda3/envs/modelscope/lib/python3.8/site-packages/funasr_onnx/utils/utils.py", line 208, in init self._verify_model(model_file) File "/home/ybZhang/miniconda3/envs/modelscope/lib/python3.8/site-packages/funasr_onnx/utils/utils.py", line 247, in _verify_model raise FileNotFoundError(f'{model_path} does not exists.') FileNotFoundError: speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404-finetune/model_eb_quant.onnx does not exists. self.ort_infer_eb = OrtInferSession(model_eb_file, device_id, intra_op_num_threads=intra_op_num_threads) File "/home/ybZhang/miniconda3/envs/modelscope/lib/python3.8/site-packages/funasr_onnx/utils/utils.py", line 208, in init self._verify_model(model_file) File "/home/ybZhang/miniconda3/envs/modelscope/lib/python3.8/site-packages/funasr_onnx/utils/utils.py", line 247, in _verify_model raise FileNotFoundError(f'{model_path} does not exists.') FileNotFoundError: speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404-finetune/model_eb_quant.onnx does not exists.

why-there commented 10 months ago

遇到一样的问题,请问这个问题最后如何解决的呢?