csukuangfj commited on
Commit
18d7e37
1 Parent(s): 0c2bc86

add a new model (zipformer + multidataset)

Browse files
Files changed (1) hide show
  1. model.py +6 -1
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 repo_id == "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11":
 
 
 
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
  }