Commit
·
0ae65b0
1
Parent(s):
9edf22d
add a new multidataset zipformer model
Browse files
model.py
CHANGED
@@ -283,6 +283,7 @@ def _get_english_model(
|
|
283 |
) -> sherpa.OfflineRecognizer:
|
284 |
assert repo_id in [
|
285 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
|
|
286 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
287 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
288 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
@@ -304,6 +305,12 @@ def _get_english_model(
|
|
304 |
):
|
305 |
filename = "cpu_jit-torch-1.10.pt"
|
306 |
|
|
|
|
|
|
|
|
|
|
|
|
|
307 |
nn_model = _get_nn_model_filename(
|
308 |
repo_id=repo_id,
|
309 |
filename=filename,
|
@@ -742,6 +749,7 @@ chinese_models = {
|
|
742 |
|
743 |
english_models = {
|
744 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
|
|
745 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
|
746 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
|
747 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|
|
|
283 |
) -> sherpa.OfflineRecognizer:
|
284 |
assert repo_id in [
|
285 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
286 |
+
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04", # noqa
|
287 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
288 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
289 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
|
|
305 |
):
|
306 |
filename = "cpu_jit-torch-1.10.pt"
|
307 |
|
308 |
+
if (
|
309 |
+
repo_id
|
310 |
+
== "yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04"
|
311 |
+
):
|
312 |
+
filename = "cpu_jit-epoch-30-avg-4.pt"
|
313 |
+
|
314 |
nn_model = _get_nn_model_filename(
|
315 |
repo_id=repo_id,
|
316 |
filename=filename,
|
|
|
749 |
|
750 |
english_models = {
|
751 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
752 |
+
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04": _get_english_model, # noqa
|
753 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
|
754 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
|
755 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|