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