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