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

6
sherpa-onnx/kotlin-api/OfflineRecognizer.kt
View File

@@ -128,8 +128,8 @@ by following the code)
@param type
0 - csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28 (Chinese)
https://k2-fsa.github.io/sherpa/onnx/pretrained_models/offline-paraformer/paraformer-models.html#csukuangfj-sherpa-onnx-paraformer-zh-2023-03-28-chinese
0 - csukuangfj/sherpa-onnx-paraformer-zh-2023-09-14 (Chinese)
https://k2-fsa.github.io/sherpa/onnx/pretrained_models/offline-paraformer/paraformer-models.html#csukuangfj-sherpa-onnx-paraformer-zh-2023-09-14-chinese
int8
1 - icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04 (English)
@@ -152,7 +152,7 @@ by following the code)
fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
when (type) {
0 -> {
val modelDir = "sherpa-onnx-paraformer-zh-2023-03-28"
val modelDir = "sherpa-onnx-paraformer-zh-2023-09-14"
return OfflineModelConfig(
paraformer = OfflineParaformerModelConfig(
model = "$modelDir/model.int8.onnx",

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",