mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
0c3c9be2c4
* onnx fp16模型 * paraformer-offline [fp32 fp16 onnx-gpu] * paraformer-offline [fp32 fp16 onnx-gpu] * Update export.py --------- Co-authored-by: zhifu gao <zhifu.gzf@alibaba-inc.com>
10 lines
420 B
Python
10 lines
420 B
Python
# method2, inference from local path
|
|
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"
|
|
)
|
|
|
|
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
|