add nemo parakeet-tdt-0.6b-v2

model.py

@@ -1270,6 +1270,52 @@ def _get_french_pre_trained_model(
     return recognizer
 
 
+@lru_cache(maxsize=10)
+def _get_sherpa_onnx_nemo_transducer_models_int8(
+    repo_id: str,
+    decoding_method: str,
+    num_active_paths: int,
+) -> sherpa_onnx.OfflineRecognizer:
+    assert repo_id in [
+        "csukuangfj/sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8",
+    ], repo_id
+
+    encoder_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="encoder.int8.onnx",
+        subfolder=".",
+    )
+
+    decoder_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="decoder.int8.onnx",
+        subfolder=".",
+    )
+
+    joiner_model = _get_nn_model_filename(
+        repo_id=repo_id,
+        filename="joiner.int8.onnx",
+        subfolder=".",
+    )
+
+    tokens = _get_token_filename(repo_id=repo_id, subfolder=".")
+
+    recognizer = sherpa_onnx.OfflineRecognizer.from_transducer(
+        tokens=tokens,
+        encoder=encoder_model,
+        decoder=decoder_model,
+        joiner=joiner_model,
+        num_threads=2,
+        sample_rate=16000,
+        feature_dim=128,  # no used
+        model_type="nemo_transducer",
+        decoding_method=decoding_method,
+        max_active_paths=num_active_paths,
+    )
+
+    return recognizer
+
+
 @lru_cache(maxsize=10)
 def _get_sherpa_onnx_nemo_transducer_models(
     repo_id: str,
@@ -1860,6 +1906,7 @@ chinese_models = {
 }
 
 english_models = {
+    "csukuangfj/sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8": _get_sherpa_onnx_nemo_transducer_models_int8,
     "whisper-tiny.en": _get_whisper_model,
     "moonshine-tiny": _get_moonshine_model,
     "moonshine-base": _get_moonshine_model,

requirements.txt

@@ -10,6 +10,6 @@ numpy<2
 
 huggingface_hub
 
-#https://huggingface.co/csukuangfj/sherpa-onnx-wheels/resolve/main/cpu/1.11.3/sherpa_onnx-1.11.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
+https://huggingface.co/csukuangfj/sherpa-onnx-wheels/resolve/main/cpu/1.11.4/sherpa_onnx-1.11.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
 
 sherpa-onnx>=1.11.3