takipipo
2025-01-16 3530688e0a1b1dfbb22dcd3324db97be5bbc0d9b
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