Update model.py
Browse files
model.py
CHANGED
|
@@ -377,38 +377,38 @@ def _get_vietnamese_pretrained_model(
|
|
| 377 |
# subfolder=".",
|
| 378 |
# )
|
| 379 |
|
| 380 |
-
decoder_model = "decoder-epoch-
|
| 381 |
|
| 382 |
-
if repo_id == "
|
| 383 |
# encoder_model = _get_nn_model_filename(
|
| 384 |
# repo_id=repo_id,
|
| 385 |
# filename="encoder-epoch-12-avg-8.int8.onnx",
|
| 386 |
# subfolder=".",
|
| 387 |
# )
|
| 388 |
|
| 389 |
-
encoder_model = "encoder-epoch-
|
| 390 |
|
| 391 |
# joiner_model = _get_nn_model_filename(
|
| 392 |
# repo_id=repo_id,
|
| 393 |
# filename="joiner-epoch-12-avg-8.int8.onnx",
|
| 394 |
# subfolder=".",
|
| 395 |
# )
|
| 396 |
-
joiner_model = "joiner-epoch-
|
| 397 |
-
elif repo_id == "
|
| 398 |
# encoder_model = _get_nn_model_filename(
|
| 399 |
# repo_id=repo_id,
|
| 400 |
# filename="encoder-epoch-12-avg-8.onnx",
|
| 401 |
# subfolder=".",
|
| 402 |
# )
|
| 403 |
|
| 404 |
-
encoder_model = "encoder-epoch-
|
| 405 |
|
| 406 |
# joiner_model = _get_nn_model_filename(
|
| 407 |
# repo_id=repo_id,
|
| 408 |
# filename="joiner-epoch-12-avg-8.onnx",
|
| 409 |
# subfolder=".",
|
| 410 |
# )
|
| 411 |
-
joiner_model = "joiner-epoch-
|
| 412 |
else:
|
| 413 |
raise ValueError(f"repo_id: {repo_id}")
|
| 414 |
|
|
|
|
| 377 |
# subfolder=".",
|
| 378 |
# )
|
| 379 |
|
| 380 |
+
decoder_model = "decoder-epoch-20-avg-10.onnx"
|
| 381 |
|
| 382 |
+
if repo_id == "hynt/sherpa-onnx-zipformer-vi-int8-2025-10-16":
|
| 383 |
# encoder_model = _get_nn_model_filename(
|
| 384 |
# repo_id=repo_id,
|
| 385 |
# filename="encoder-epoch-12-avg-8.int8.onnx",
|
| 386 |
# subfolder=".",
|
| 387 |
# )
|
| 388 |
|
| 389 |
+
encoder_model = "encoder-epoch-20-avg-10.int8.onnx"
|
| 390 |
|
| 391 |
# joiner_model = _get_nn_model_filename(
|
| 392 |
# repo_id=repo_id,
|
| 393 |
# filename="joiner-epoch-12-avg-8.int8.onnx",
|
| 394 |
# subfolder=".",
|
| 395 |
# )
|
| 396 |
+
joiner_model = "joiner-epoch-20-avg-10.int8.onnx"
|
| 397 |
+
elif repo_id == "hynt/sherpa-onnx-zipformer-vi-2025-10-16":
|
| 398 |
# encoder_model = _get_nn_model_filename(
|
| 399 |
# repo_id=repo_id,
|
| 400 |
# filename="encoder-epoch-12-avg-8.onnx",
|
| 401 |
# subfolder=".",
|
| 402 |
# )
|
| 403 |
|
| 404 |
+
encoder_model = "encoder-epoch-20-avg-10.onnx"
|
| 405 |
|
| 406 |
# joiner_model = _get_nn_model_filename(
|
| 407 |
# repo_id=repo_id,
|
| 408 |
# filename="joiner-epoch-12-avg-8.onnx",
|
| 409 |
# subfolder=".",
|
| 410 |
# )
|
| 411 |
+
joiner_model = "joiner-epoch-20-avg-10.onnx"
|
| 412 |
else:
|
| 413 |
raise ValueError(f"repo_id: {repo_id}")
|
| 414 |
|