Commit
·
18d7e37
1
Parent(s):
0c2bc86
add a new model (zipformer + multidataset)
Browse files
model.py
CHANGED
@@ -124,10 +124,14 @@ def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
|
124 |
assert repo_id in [
|
125 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
126 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
|
|
127 |
], repo_id
|
128 |
|
129 |
filename = "cpu_jit.pt"
|
130 |
-
if
|
|
|
|
|
|
|
131 |
filename = "cpu_jit-torch-1.10.0.pt"
|
132 |
|
133 |
nn_model_filename = _get_nn_model_filename(
|
@@ -239,6 +243,7 @@ chinese_models = {
|
|
239 |
|
240 |
english_models = {
|
241 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
|
|
242 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model, # noqa
|
243 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model, # noqa
|
244 |
}
|
|
|
124 |
assert repo_id in [
|
125 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
126 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
127 |
+
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
128 |
], repo_id
|
129 |
|
130 |
filename = "cpu_jit.pt"
|
131 |
+
if (
|
132 |
+
repo_id
|
133 |
+
== "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11"
|
134 |
+
):
|
135 |
filename = "cpu_jit-torch-1.10.0.pt"
|
136 |
|
137 |
nn_model_filename = _get_nn_model_filename(
|
|
|
243 |
|
244 |
english_models = {
|
245 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
246 |
+
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_librispeech_pre_trained_model, # noqa
|
247 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model, # noqa
|
248 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model, # noqa
|
249 |
}
|