csukuangfj
commited on
Commit
•
451aa75
1
Parent(s):
6c15050
update model from ricky
Browse files
model.py
CHANGED
@@ -124,6 +124,7 @@ def _get_gigaspeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
|
124 |
@lru_cache(maxsize=10)
|
125 |
def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
126 |
assert repo_id in [
|
|
|
127 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
128 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
129 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
@@ -136,6 +137,12 @@ def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
|
136 |
):
|
137 |
filename = "cpu_jit-torch-1.10.0.pt"
|
138 |
|
|
|
|
|
|
|
|
|
|
|
|
|
139 |
nn_model_filename = _get_nn_model_filename(
|
140 |
repo_id=repo_id,
|
141 |
filename=filename,
|
@@ -266,6 +273,7 @@ chinese_models = {
|
|
266 |
|
267 |
english_models = {
|
268 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
|
|
269 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_librispeech_pre_trained_model, # noqa
|
270 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model, # noqa
|
271 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model, # noqa
|
|
|
124 |
@lru_cache(maxsize=10)
|
125 |
def _get_librispeech_pre_trained_model(repo_id: str) -> OfflineAsr:
|
126 |
assert repo_id in [
|
127 |
+
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
128 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
129 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
130 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
|
|
137 |
):
|
138 |
filename = "cpu_jit-torch-1.10.0.pt"
|
139 |
|
140 |
+
if (
|
141 |
+
repo_id
|
142 |
+
== "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02"
|
143 |
+
):
|
144 |
+
filename = "cpu_jit-torch-1.10.pt"
|
145 |
+
|
146 |
nn_model_filename = _get_nn_model_filename(
|
147 |
repo_id=repo_id,
|
148 |
filename=filename,
|
|
|
273 |
|
274 |
english_models = {
|
275 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
276 |
+
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_librispeech_pre_trained_model, # noqa
|
277 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_librispeech_pre_trained_model, # noqa
|
278 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_librispeech_pre_trained_model, # noqa
|
279 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_librispeech_pre_trained_model, # noqa
|