csukuangfj commited on
Commit
06f3af4
·
1 Parent(s): 0986216

add more models

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