File size: 750 Bytes
e4406a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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) |