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

Support specifying providers in Python API (#198)

Browse Source
This commit is contained in:
Fangjun Kuang
2023-07-06 10:14:01 +08:00
committed by GitHub
parent 3a08191a43
commit 33bf8dc1f4
4 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -72,6 +72,7 @@ class TestOfflineRecognizer(unittest.TestCase):
joiner=joiner,
tokens=tokens,
num_threads=1,
provider="cpu",
)
s = recognizer.create_stream()
@@ -106,6 +107,7 @@ class TestOfflineRecognizer(unittest.TestCase):
joiner=joiner,
tokens=tokens,
num_threads=1,
provider="cpu",
)
s0 = recognizer.create_stream()
@@ -143,6 +145,7 @@ class TestOfflineRecognizer(unittest.TestCase):
paraformer=model,
tokens=tokens,
num_threads=1,
provider="cpu",
)
s = recognizer.create_stream()
@@ -172,6 +175,7 @@ class TestOfflineRecognizer(unittest.TestCase):
paraformer=model,
tokens=tokens,
num_threads=1,
provider="cpu",
)
s0 = recognizer.create_stream()
@@ -214,6 +218,7 @@ class TestOfflineRecognizer(unittest.TestCase):
model=model,
tokens=tokens,
num_threads=1,
provider="cpu",
)
s = recognizer.create_stream()
@@ -242,6 +247,7 @@ class TestOfflineRecognizer(unittest.TestCase):
model=model,
tokens=tokens,
num_threads=1,
provider="cpu",
)
s0 = recognizer.create_stream()

2
sherpa-onnx/python/tests/test_online_recognizer.py
View File

@@ -72,6 +72,7 @@ class TestOnlineRecognizer(unittest.TestCase):
tokens=tokens,
num_threads=1,
decoding_method=decoding_method,
provider="cpu",
)
s = recognizer.create_stream()
samples, sample_rate = read_wave(wave0)
@@ -115,6 +116,7 @@ class TestOnlineRecognizer(unittest.TestCase):
tokens=tokens,
num_threads=1,
decoding_method=decoding_method,
provider="cpu",
)
streams = []
waves = [wave0, wave1, wave2, wave3, wave4]