FunASR

mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00

History

游雁 d377874967 export model		2023-02-08 10:19:07 +08:00
..
models	export model	2023-02-07 22:51:39 +08:00
utils	export model	2023-02-07 17:45:59 +08:00
__init__.py	export model	2023-02-07 15:19:18 +08:00
export_model.py	export model	2023-02-07 22:51:39 +08:00
README.md	export model	2023-02-08 10:19:07 +08:00
test_onnx.py	export model test	2023-02-07 22:57:08 +08:00
test_torchscripts.py	export model test	2023-02-07 22:57:08 +08:00

README.md

environment

funasr 0.1.7
python 3.7
torch 1.11.0
modelscope 1.2.0

install modelscope and funasr

The install is the same as funasr

export onnx format model

Export model from modelscope

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
export_model.export_from_modelscope('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

Export model from local path

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
export_model.export_from_local('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

export torchscripts format model

Export model from modelscope

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=False)
export_model.export_from_modelscope('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

Export model from local path

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=False)
export_model.export_from_local('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')