Commit
·
a6f60a8
1
Parent(s):
5b99533
Add a new model
Browse files
model.py
CHANGED
|
@@ -182,6 +182,10 @@ def get_pretrained_model(
|
|
| 182 |
return chinese_models[repo_id](
|
| 183 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
| 184 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 185 |
elif repo_id in chinese_dialect_models:
|
| 186 |
return chinese_dialect_models[repo_id](
|
| 187 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
|
@@ -2008,6 +2012,7 @@ def _get_paraformer_pre_trained_model(
|
|
| 2008 |
"csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09",
|
| 2009 |
"csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en",
|
| 2010 |
"csukuangfj/sherpa-onnx-paraformer-en-2024-03-09",
|
|
|
|
| 2011 |
], repo_id
|
| 2012 |
|
| 2013 |
nn_model = _get_nn_model_filename(
|
|
@@ -2147,6 +2152,10 @@ chinese_dialect_models = {
|
|
| 2147 |
"csukuangfj/sherpa-onnx-telespeech-ctc-int8-zh-2024-06-04": _get_chinese_dialect_models,
|
| 2148 |
}
|
| 2149 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2150 |
chinese_models = {
|
| 2151 |
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
| 2152 |
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
|
@@ -2319,6 +2328,7 @@ portuguese_brazlian_models = {
|
|
| 2319 |
all_models = {
|
| 2320 |
**twenty_five_languages_models,
|
| 2321 |
**multi_lingual_models,
|
|
|
|
| 2322 |
**chinese_models,
|
| 2323 |
**english_models,
|
| 2324 |
**chinese_english_mixed_models,
|
|
@@ -2344,6 +2354,7 @@ language_to_models = {
|
|
| 2344 |
"超多种中文方言": list(chinese_dialect_models.keys()),
|
| 2345 |
"Chinese": list(chinese_models.keys()),
|
| 2346 |
"English": list(english_models.keys()),
|
|
|
|
| 2347 |
"Chinese+English": list(chinese_english_mixed_models.keys()),
|
| 2348 |
"Chinese+English+Cantonese": list(chinese_cantonese_english_models.keys()),
|
| 2349 |
"Chinese+English+Cantonese+Japanese+Korean": list(
|
|
|
|
| 182 |
return chinese_models[repo_id](
|
| 183 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
| 184 |
)
|
| 185 |
+
elif repo_id in sichuan_models:
|
| 186 |
+
return sichuan_models[repo_id](
|
| 187 |
+
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
| 188 |
+
)
|
| 189 |
elif repo_id in chinese_dialect_models:
|
| 190 |
return chinese_dialect_models[repo_id](
|
| 191 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
|
|
|
| 2012 |
"csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09",
|
| 2013 |
"csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en",
|
| 2014 |
"csukuangfj/sherpa-onnx-paraformer-en-2024-03-09",
|
| 2015 |
+
"csukuangfj/sherpa-onnx-paraformer-zh-int8-2025-10-07",
|
| 2016 |
], repo_id
|
| 2017 |
|
| 2018 |
nn_model = _get_nn_model_filename(
|
|
|
|
| 2152 |
"csukuangfj/sherpa-onnx-telespeech-ctc-int8-zh-2024-06-04": _get_chinese_dialect_models,
|
| 2153 |
}
|
| 2154 |
|
| 2155 |
+
sichuan_models = {
|
| 2156 |
+
"csukuangfj/sherpa-onnx-paraformer-zh-int8-2025-10-07": _get_paraformer_pre_trained_model,
|
| 2157 |
+
}
|
| 2158 |
+
|
| 2159 |
chinese_models = {
|
| 2160 |
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-int8-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
| 2161 |
"csukuangfj/sherpa-onnx-streaming-zipformer-ctc-zh-2025-06-30": _get_streaming_zipformer_ctc_pre_trained_model,
|
|
|
|
| 2328 |
all_models = {
|
| 2329 |
**twenty_five_languages_models,
|
| 2330 |
**multi_lingual_models,
|
| 2331 |
+
**sichuan_models,
|
| 2332 |
**chinese_models,
|
| 2333 |
**english_models,
|
| 2334 |
**chinese_english_mixed_models,
|
|
|
|
| 2354 |
"超多种中文方言": list(chinese_dialect_models.keys()),
|
| 2355 |
"Chinese": list(chinese_models.keys()),
|
| 2356 |
"English": list(english_models.keys()),
|
| 2357 |
+
"四川话": list(sichuan_models.keys()),
|
| 2358 |
"Chinese+English": list(chinese_english_mixed_models.keys()),
|
| 2359 |
"Chinese+English+Cantonese": list(chinese_cantonese_english_models.keys()),
|
| 2360 |
"Chinese+English+Cantonese+Japanese+Korean": list(
|