From c4412495bded97f40c7cdd5ab37cb5f789bfff41 Mon Sep 17 00:00:00 2001 From: inisis Date: Tue, 17 Dec 2024 12:40:04 +0800 Subject: [PATCH] add onnxslim reference (#2314) --- README_zh.md | 6 ++++++ runtime/python/utils/test_rtf.py | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/README_zh.md b/README_zh.md index b90f0e4c9..272ab13d5 100644 --- a/README_zh.md +++ b/README_zh.md @@ -312,6 +312,12 @@ model = AutoModel(model="paraformer") res = model.export(quantize=False) ``` +### 优化onnx +```shell +# pip3 install -U onnxslim +onnxslim model.onnx model.onnx +``` + ### 测试ONNX ```python # pip3 install -U funasr-onnx diff --git a/runtime/python/utils/test_rtf.py b/runtime/python/utils/test_rtf.py index 112025d6d..8e50c34f8 100644 --- a/runtime/python/utils/test_rtf.py +++ b/runtime/python/utils/test_rtf.py @@ -1,7 +1,7 @@ import time import sys import librosa -from funasr.utils.types import str2bool +from funasr.utils.type_utils import str2bool import argparse