csukuangfj commited on
Commit
6b92718
·
1 Parent(s): 7d508be

fix zipformer ctc

Browse files
Files changed (1) hide show
  1. model.py +36 -0
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,