export.py
@@ -2,8 +2,9 @@ from funasr import AutoModel model = AutoModel( model="/raid/t3cv/wangch/WORK_SAPCE/ASR/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" model="iic/emotion2vec_base", hub="ms" ) res = model.export(type="onnx", quantize=False, opset_version=13, device='cuda') # fp32 onnx-gpu res = model.export(type="onnx", quantize=False, opset_version=13, device='cpu') # fp32 onnx-gpu # res = model.export(type="onnx_fp16", quantize=False, opset_version=13, device='cuda') # fp16 onnx-gpu
