TestTranslator / scripts /export_onnx.py
yujuanqin's picture
add asr test
db0d138
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)
model_dir = "/Users/jeqin/work/code/Fun-ASR-Nano-2512"
model = AutoModel(model=model_dir, trust_remote_code=True, remote_code="./model.py")
model_dir = model.export(type="onnx", quantize=True, disable_update=True)