csukuangfj commited on
Commit
9ff62ad
·
1 Parent(s): f74b364

add more models

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