Commit
·
06f3af4
1
Parent(s):
0986216
add more models
Browse files
model.py
CHANGED
@@ -1469,6 +1469,60 @@ def _get_sherpa_onnx_offline_zipformer_pre_trained_model(
|
|
1469 |
return recognizer
|
1470 |
|
1471 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1472 |
@lru_cache(maxsize=10)
|
1473 |
def _get_streaming_zipformer_pre_trained_model(
|
1474 |
repo_id: str,
|
@@ -1889,6 +1943,11 @@ chinese_dialect_models = {
|
|
1889 |
}
|
1890 |
|
1891 |
chinese_models = {
|
|
|
|
|
|
|
|
|
|
|
1892 |
"csukuangfj/sherpa-onnx-paraformer-zh-2024-03-09": _get_paraformer_pre_trained_model,
|
1893 |
"luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model, # noqa
|
1894 |
"csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09": _get_paraformer_pre_trained_model,
|
|
|
1469 |
return recognizer
|
1470 |
|
1471 |
|
1472 |
+
@lru_cache(maxsize=10)
|
1473 |
+
def _get_streaming_zipformer_ctc_pre_trained_model(
|
1474 |
+
repo_id: str,
|
1475 |
+
decoding_method: str,
|
1476 |
+
num_active_paths: int,
|
1477 |
+
) -> sherpa_onnx.OnlineRecognizer:
|
1478 |
+
assert repo_id in [
|
1479 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30",
|
1480 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30",
|
1481 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-fp16-2025-06-30",
|
1482 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-xlarge-int8-2025-06-30",
|
1483 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-xlarge-fp16-2025-06-30",
|
1484 |
+
], repo_id
|
1485 |
+
|
1486 |
+
if repo_id in (
|
1487 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30",
|
1488 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-xlarge-int8-2025-06-30",
|
1489 |
+
):
|
1490 |
+
model = _get_nn_model_filename(
|
1491 |
+
repo_id=repo_id,
|
1492 |
+
filename="model.int8.onnx",
|
1493 |
+
subfolder=".",
|
1494 |
+
)
|
1495 |
+
elif repo_id in (
|
1496 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-fp16-2025-06-30",
|
1497 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-xlarge-fp16-2025-06-30",
|
1498 |
+
):
|
1499 |
+
model = _get_nn_model_filename(
|
1500 |
+
repo_id=repo_id,
|
1501 |
+
filename="model.fp16.onnx",
|
1502 |
+
subfolder=".",
|
1503 |
+
)
|
1504 |
+
elif repo_id in ("csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30",):
|
1505 |
+
model = _get_nn_model_filename(
|
1506 |
+
repo_id=repo_id,
|
1507 |
+
filename="model.onnx",
|
1508 |
+
subfolder=".",
|
1509 |
+
)
|
1510 |
+
|
1511 |
+
tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
|
1512 |
+
|
1513 |
+
recognizer = sherpa_onnx.OnlineRecognizer.from_zipformer2_ctc(
|
1514 |
+
tokens=tokens,
|
1515 |
+
model=model,
|
1516 |
+
num_threads=2,
|
1517 |
+
sample_rate=16000,
|
1518 |
+
feature_dim=80,
|
1519 |
+
decoding_method=decoding_method,
|
1520 |
+
max_active_paths=num_active_paths,
|
1521 |
+
)
|
1522 |
+
|
1523 |
+
return recognizer
|
1524 |
+
|
1525 |
+
|
1526 |
@lru_cache(maxsize=10)
|
1527 |
def _get_streaming_zipformer_pre_trained_model(
|
1528 |
repo_id: str,
|
|
|
1943 |
}
|
1944 |
|
1945 |
chinese_models = {
|
1946 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
1947 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
1948 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-fp16-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
1949 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-xlarge-int8-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
1950 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-xlarge-fp16-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
1951 |
"csukuangfj/sherpa-onnx-paraformer-zh-2024-03-09": _get_paraformer_pre_trained_model,
|
1952 |
"luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model, # noqa
|
1953 |
"csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09": _get_paraformer_pre_trained_model,
|