Commit
·
7fd804c
1
Parent(s):
3f09dbf
add a new english model
Browse files
model.py
CHANGED
|
@@ -1586,25 +1586,45 @@ def _get_streaming_zipformer_pre_trained_model(
|
|
| 1586 |
assert repo_id in [
|
| 1587 |
"csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
|
| 1588 |
"k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16",
|
|
|
|
| 1589 |
], repo_id
|
| 1590 |
|
| 1591 |
-
|
| 1592 |
-
|
| 1593 |
-
|
| 1594 |
-
|
| 1595 |
-
|
|
|
|
| 1596 |
|
| 1597 |
-
|
| 1598 |
-
|
| 1599 |
-
|
| 1600 |
-
|
| 1601 |
-
|
| 1602 |
|
| 1603 |
-
|
| 1604 |
-
|
| 1605 |
-
|
| 1606 |
-
|
| 1607 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1608 |
|
| 1609 |
tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
|
| 1610 |
|
|
@@ -2030,6 +2050,7 @@ english_models = {
|
|
| 2030 |
"whisper-small.en": _get_whisper_model,
|
| 2031 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_ctc_110m-en-36000": _get_sherpa_onnx_nemo_ctc_models,
|
| 2032 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_transducer_110m-en-36000": _get_sherpa_onnx_nemo_transducer_models,
|
|
|
|
| 2033 |
# "whisper-medium.en": _get_whisper_model,
|
| 2034 |
"csukuangfj/sherpa-onnx-zipformer-en-libriheavy-20230926-large": _get_sherpa_onnx_offline_zipformer_pre_trained_model,
|
| 2035 |
"csukuangfj/sherpa-onnx-zipformer-en-libriheavy-20230926-medium": _get_sherpa_onnx_offline_zipformer_pre_trained_model,
|
|
|
|
| 1586 |
assert repo_id in [
|
| 1587 |
"csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
|
| 1588 |
"k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16",
|
| 1589 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
|
| 1590 |
], repo_id
|
| 1591 |
|
| 1592 |
+
if repo_id == "csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06":
|
| 1593 |
+
encoder_model = _get_nn_model_filename(
|
| 1594 |
+
repo_id=repo_id,
|
| 1595 |
+
filename="encoder.onnx",
|
| 1596 |
+
subfolder=".",
|
| 1597 |
+
)
|
| 1598 |
|
| 1599 |
+
decoder_model = _get_nn_model_filename(
|
| 1600 |
+
repo_id=repo_id,
|
| 1601 |
+
filename="decoder.onnx",
|
| 1602 |
+
subfolder=".",
|
| 1603 |
+
)
|
| 1604 |
|
| 1605 |
+
joiner_model = _get_nn_model_filename(
|
| 1606 |
+
repo_id=repo_id,
|
| 1607 |
+
filename="joiner.onnx",
|
| 1608 |
+
subfolder=".",
|
| 1609 |
+
)
|
| 1610 |
+
else:
|
| 1611 |
+
encoder_model = _get_nn_model_filename(
|
| 1612 |
+
repo_id=repo_id,
|
| 1613 |
+
filename="encoder-epoch-99-avg-1.onnx",
|
| 1614 |
+
subfolder=".",
|
| 1615 |
+
)
|
| 1616 |
+
|
| 1617 |
+
decoder_model = _get_nn_model_filename(
|
| 1618 |
+
repo_id=repo_id,
|
| 1619 |
+
filename="decoder-epoch-99-avg-1.onnx",
|
| 1620 |
+
subfolder=".",
|
| 1621 |
+
)
|
| 1622 |
+
|
| 1623 |
+
joiner_model = _get_nn_model_filename(
|
| 1624 |
+
repo_id=repo_id,
|
| 1625 |
+
filename="joiner-epoch-99-avg-1.onnx",
|
| 1626 |
+
subfolder=".",
|
| 1627 |
+
)
|
| 1628 |
|
| 1629 |
tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
|
| 1630 |
|
|
|
|
| 2050 |
"whisper-small.en": _get_whisper_model,
|
| 2051 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_ctc_110m-en-36000": _get_sherpa_onnx_nemo_ctc_models,
|
| 2052 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_transducer_110m-en-36000": _get_sherpa_onnx_nemo_transducer_models,
|
| 2053 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06": _get_streaming_zipformer_pre_trained_model,
|
| 2054 |
# "whisper-medium.en": _get_whisper_model,
|
| 2055 |
"csukuangfj/sherpa-onnx-zipformer-en-libriheavy-20230926-large": _get_sherpa_onnx_offline_zipformer_pre_trained_model,
|
| 2056 |
"csukuangfj/sherpa-onnx-zipformer-en-libriheavy-20230926-medium": _get_sherpa_onnx_offline_zipformer_pre_trained_model,
|