Commit
·
6b92718
1
Parent(s):
7d508be
fix zipformer ctc
Browse files
model.py
CHANGED
@@ -1530,6 +1530,42 @@ def _get_streaming_zipformer_ctc_pre_trained_model(
|
|
1530 |
return recognizer
|
1531 |
|
1532 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1533 |
@lru_cache(maxsize=10)
|
1534 |
def _get_streaming_zipformer_pre_trained_model(
|
1535 |
repo_id: str,
|
|
|
1530 |
return recognizer
|
1531 |
|
1532 |
|
1533 |
+
@lru_cache(maxsize=10)
|
1534 |
+
def _get_non_streaming_zipformer_ctc_pre_trained_model(
|
1535 |
+
repo_id: str,
|
1536 |
+
decoding_method: str,
|
1537 |
+
num_active_paths: int,
|
1538 |
+
) -> sherpa_onnx.OfflineRecognizer:
|
1539 |
+
assert repo_id in [
|
1540 |
+
"csukuangfj/sherpa-onnx-zipformer-ctc-zh-int8-2025-07-03",
|
1541 |
+
"csukuangfj/sherpa-onnx-zipformer-ctc-zh-2025-07-03",
|
1542 |
+
], repo_id
|
1543 |
+
|
1544 |
+
if "int8" in repo_id:
|
1545 |
+
model = _get_nn_model_filename(
|
1546 |
+
repo_id=repo_id,
|
1547 |
+
filename="model.int8.onnx",
|
1548 |
+
subfolder=".",
|
1549 |
+
)
|
1550 |
+
else:
|
1551 |
+
model = _get_nn_model_filename(
|
1552 |
+
repo_id=repo_id,
|
1553 |
+
filename="model.onnx",
|
1554 |
+
subfolder=".",
|
1555 |
+
)
|
1556 |
+
tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
|
1557 |
+
|
1558 |
+
recognizer = sherpa_onnx.OfflineRecognizer.from_zipformer_ctc(
|
1559 |
+
tokens=tokens,
|
1560 |
+
model=model,
|
1561 |
+
num_threads=2,
|
1562 |
+
sample_rate=16000,
|
1563 |
+
feature_dim=80,
|
1564 |
+
)
|
1565 |
+
|
1566 |
+
return recognizer
|
1567 |
+
|
1568 |
+
|
1569 |
@lru_cache(maxsize=10)
|
1570 |
def _get_streaming_zipformer_pre_trained_model(
|
1571 |
repo_id: str,
|