csukuangfj commited on
Commit
451aa75
1 Parent(s): 6c15050

update model from ricky

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