csukuangfj commited on
Commit
0ae65b0
1 Parent(s): 9edf22d

add a new multidataset zipformer model

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