TestTranslator / scripts /export_onnx.py
yujuanqin's picture
update scripts and test_data
e4406a3
from funasr import AutoModel
model_dir = "/Users/moyoyo/code/Translator/moyoyo_asr_models"
asr_model_path = model_dir + '/speech_seaco_paraformer_large_asr_nat-zh-cn-16k-common-vocab8404-pytorch'
vad_model_path = model_dir + '/speech_fsmn_vad_zh-cn-16k-common-pytorch'
punc_model_path = model_dir + '/punc_ct-transformer_cn-en-common-vocab471067-large'
model = AutoModel(model=asr_model_path)
model_dir = model.export(type="onnx", quantize=True, disable_update=True)
print(model_dir)
model = AutoModel(model=vad_model_path)
model_dir = model.export(type="onnx", quantize=True, disable_update=True)
print(model_dir)
model = AutoModel(model=punc_model_path)
model_dir = model.export(type="onnx", quantize=True, disable_update=True)
print(model_dir)