FunASR/export.py
Thomas Young 46721dc84a
fix export error if using cpu device; fix model path (#2265)
Co-authored-by: siyuan.yang <siyuan.yang@sophgo.com>
2024-12-05 11:01:29 +08:00

10 lines
384 B
Python

# method2, inference from local path
from funasr import AutoModel
model = AutoModel(
model="iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch"
)
res = model.export(type="onnx", quantize=False, opset_version=13, device='cuda') # fp32 onnx-gpu
# res = model.export(type="onnx_fp16", quantize=False, opset_version=13, device='cuda') # fp16 onnx-gpu