Commit
·
9ff62ad
1
Parent(s):
f74b364
add more models
Browse files
model.py
CHANGED
|
@@ -1282,6 +1282,7 @@ def _get_sherpa_onnx_nemo_transducer_models_int8(
|
|
| 1282 |
) -> sherpa_onnx.OfflineRecognizer:
|
| 1283 |
assert repo_id in [
|
| 1284 |
"csukuangfj/sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8",
|
|
|
|
| 1285 |
], repo_id
|
| 1286 |
|
| 1287 |
encoder_model = _get_nn_model_filename(
|
|
@@ -1328,6 +1329,7 @@ def _get_sherpa_onnx_nemo_transducer_models(
|
|
| 1328 |
) -> sherpa_onnx.OfflineRecognizer:
|
| 1329 |
assert repo_id in [
|
| 1330 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_transducer_110m-en-36000",
|
|
|
|
| 1331 |
], repo_id
|
| 1332 |
|
| 1333 |
encoder_model = _get_nn_model_filename(
|
|
@@ -1376,6 +1378,8 @@ def _get_sherpa_onnx_nemo_ctc_models(
|
|
| 1376 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_ctc_110m-en-36000",
|
| 1377 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc",
|
| 1378 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc-int8",
|
|
|
|
|
|
|
| 1379 |
], repo_id
|
| 1380 |
|
| 1381 |
if "int8" in repo_id:
|
|
@@ -2075,6 +2079,10 @@ arabic_models = {
|
|
| 2075 |
|
| 2076 |
german_models = {
|
| 2077 |
"csukuangfj/wav2vec2.0-torchaudio": _get_german_pre_trained_model,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2078 |
}
|
| 2079 |
|
| 2080 |
french_models = {
|
|
@@ -2126,6 +2134,8 @@ vietnamese_models = {
|
|
| 2126 |
portuguese_brazlian_models = {
|
| 2127 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc": _get_sherpa_onnx_nemo_ctc_models,
|
| 2128 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc-int8": _get_sherpa_onnx_nemo_ctc_models,
|
|
|
|
|
|
|
| 2129 |
}
|
| 2130 |
|
| 2131 |
|
|
|
|
| 1282 |
) -> sherpa_onnx.OfflineRecognizer:
|
| 1283 |
assert repo_id in [
|
| 1284 |
"csukuangfj/sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8",
|
| 1285 |
+
"csukuangfj/sherpa-onnx-nemo-transducer-stt_de_fastconformer_hybrid_large_pc-int8",
|
| 1286 |
], repo_id
|
| 1287 |
|
| 1288 |
encoder_model = _get_nn_model_filename(
|
|
|
|
| 1329 |
) -> sherpa_onnx.OfflineRecognizer:
|
| 1330 |
assert repo_id in [
|
| 1331 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_transducer_110m-en-36000",
|
| 1332 |
+
"csukuangfj/sherpa-onnx-nemo-transducer-stt_de_fastconformer_hybrid_large_pc",
|
| 1333 |
], repo_id
|
| 1334 |
|
| 1335 |
encoder_model = _get_nn_model_filename(
|
|
|
|
| 1378 |
"csukuangfj/sherpa-onnx-nemo-parakeet_tdt_ctc_110m-en-36000",
|
| 1379 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc",
|
| 1380 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc-int8",
|
| 1381 |
+
"csukuangfj/sherpa-onnx-nemo-stt_de_fastconformer_hybrid_large_pc",
|
| 1382 |
+
"csukuangfj/sherpa-onnx-nemo-stt_de_fastconformer_hybrid_large_pc-int8",
|
| 1383 |
], repo_id
|
| 1384 |
|
| 1385 |
if "int8" in repo_id:
|
|
|
|
| 2079 |
|
| 2080 |
german_models = {
|
| 2081 |
"csukuangfj/wav2vec2.0-torchaudio": _get_german_pre_trained_model,
|
| 2082 |
+
"csukuangfj/sherpa-onnx-nemo-transducer-stt_de_fastconformer_hybrid_large_pc": _get_sherpa_onnx_nemo_transducer_models,
|
| 2083 |
+
"csukuangfj/sherpa-onnx-nemo-transducer-stt_de_fastconformer_hybrid_large_pc-int8": _get_sherpa_onnx_nemo_transducer_models_int8,
|
| 2084 |
+
"csukuangfj/sherpa-onnx-nemo-stt_de_fastconformer_hybrid_large_pc": _get_sherpa_onnx_nemo_ctc_models,
|
| 2085 |
+
"csukuangfj/sherpa-onnx-nemo-stt_de_fastconformer_hybrid_large_pc-int8": _get_sherpa_onnx_nemo_ctc_models,
|
| 2086 |
}
|
| 2087 |
|
| 2088 |
french_models = {
|
|
|
|
| 2134 |
portuguese_brazlian_models = {
|
| 2135 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc": _get_sherpa_onnx_nemo_ctc_models,
|
| 2136 |
"csukuangfj/sherpa-onnx-nemo-stt_pt_fastconformer_hybrid_large_pc-int8": _get_sherpa_onnx_nemo_ctc_models,
|
| 2137 |
+
"csukuangfj/sherpa-onnx-nemo-transducer-stt_pt_fastconformer_hybrid_large_pc": _get_sherpa_onnx_nemo_transducer_models,
|
| 2138 |
+
"csukuangfj/sherpa-onnx-nemo-transducer-stt_pt_fastconformer_hybrid_large_pc-int8": _get_sherpa_onnx_nemo_transducer_models_int8,
|
| 2139 |
}
|
| 2140 |
|
| 2141 |
|