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