Commit
·
ae854f0
1
Parent(s):
4a82406
minor fixes
Browse files
model.py
CHANGED
@@ -273,12 +273,9 @@ def _get_aishell2_pretrained_model(
|
|
273 |
def _get_whisper_model(
|
274 |
repo_id: str, decoding_method: str, num_active_paths: int
|
275 |
) -> sherpa_onnx.OfflineRecognizer:
|
276 |
-
print(f"get {repo_id}")
|
277 |
name = repo_id.split("-")[1]
|
278 |
assert name in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
|
279 |
-
print(f"name: {name}")
|
280 |
full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
|
281 |
-
print(f"full repo id: {full_repo_id}")
|
282 |
encoder = _get_nn_model_filename(
|
283 |
repo_id=full_repo_id,
|
284 |
filename=f"{name}-encoder.int8.ort",
|
@@ -295,15 +292,12 @@ def _get_whisper_model(
|
|
295 |
repo_id=full_repo_id, subfolder=".", filename=f"{name}-tokens.txt"
|
296 |
)
|
297 |
|
298 |
-
print("creating recognizer")
|
299 |
-
|
300 |
recognizer = sherpa_onnx.OfflineRecognizer.from_whisper(
|
301 |
encoder=encoder,
|
302 |
decoder=decoder,
|
303 |
tokens=tokens,
|
304 |
num_threads=2,
|
305 |
)
|
306 |
-
print("created recognizer")
|
307 |
|
308 |
return recognizer
|
309 |
|
|
|
273 |
def _get_whisper_model(
|
274 |
repo_id: str, decoding_method: str, num_active_paths: int
|
275 |
) -> sherpa_onnx.OfflineRecognizer:
|
|
|
276 |
name = repo_id.split("-")[1]
|
277 |
assert name in ("tiny.en", "base.en", "small.en", "medium.en"), repo_id
|
|
|
278 |
full_repo_id = "csukuangfj/sherpa-onnx-whisper-" + name
|
|
|
279 |
encoder = _get_nn_model_filename(
|
280 |
repo_id=full_repo_id,
|
281 |
filename=f"{name}-encoder.int8.ort",
|
|
|
292 |
repo_id=full_repo_id, subfolder=".", filename=f"{name}-tokens.txt"
|
293 |
)
|
294 |
|
|
|
|
|
295 |
recognizer = sherpa_onnx.OfflineRecognizer.from_whisper(
|
296 |
encoder=encoder,
|
297 |
decoder=decoder,
|
298 |
tokens=tokens,
|
299 |
num_threads=2,
|
300 |
)
|
|
|
301 |
|
302 |
return recognizer
|
303 |
|