王梦迪
2025-05-26 9038340be707aa7cbf62fcbf33ab615bb266abdb
export.py
@@ -2,8 +2,9 @@
from funasr import AutoModel
model = AutoModel(
    model="iic/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