EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Support onnxruntime 1.18.0 (#906)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-07-10 17:05:26 +08:00
committed by GitHub
parent 9e446b8501
commit dd0ff2ca06
107 changed files with 644 additions and 601 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
sherpa-onnx/python/tests/test_offline_recognizer.py
View File

@@ -130,12 +130,12 @@ class TestOfflineRecognizer(unittest.TestCase):
def test_paraformer_single_file(self):
for use_int8 in [True, False]:
if use_int8:
model = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/model.int8.onnx"
model = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/model.int8.onnx"
else:
model = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/model.onnx"
model = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/model.onnx"
tokens = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/tokens.txt"
wave0 = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/test_wavs/0.wav"
tokens = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/tokens.txt"
wave0 = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/test_wavs/0.wav"
if not Path(model).is_file():
print("skipping test_paraformer_single_file()")
@@ -157,15 +157,15 @@ class TestOfflineRecognizer(unittest.TestCase):
def test_paraformer_multiple_files(self):
for use_int8 in [True, False]:
if use_int8:
model = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/model.int8.onnx"
model = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/model.int8.onnx"
else:
model = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/model.onnx"
model = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/model.onnx"
tokens = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/tokens.txt"
wave0 = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/test_wavs/0.wav"
wave1 = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/test_wavs/1.wav"
wave2 = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/test_wavs/2.wav"
wave3 = f"{d}/sherpa-onnx-paraformer-zh-2023-03-28/test_wavs/8k.wav"
tokens = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/tokens.txt"
wave0 = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/test_wavs/0.wav"
wave1 = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/test_wavs/1.wav"
wave2 = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/test_wavs/2.wav"
wave3 = f"{d}/sherpa-onnx-paraformer-zh-2023-09-14/test_wavs/8k.wav"
if not Path(model).is_file():
print("skipping test_paraformer_multiple_files()")
@@ -267,7 +267,7 @@ class TestOfflineRecognizer(unittest.TestCase):
print(s1.result.text)
print(s2.result.text)
def test_wenet_ctc(self):
def _test_wenet_ctc(self):
models = [
"sherpa-onnx-zh-wenet-aishell",
"sherpa-onnx-zh-wenet-aishell2",