csukuangfj commited on
Commit
fc7e843
1 Parent(s): e6d227e

add a new model (fine-tuning)

Browse files
Files changed (1) hide show
  1. model.py +8 -0
model.py CHANGED
@@ -313,6 +313,7 @@ def _get_english_model(
313
  assert repo_id in [
314
  "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
315
  "yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04", # noqa
 
316
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
317
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
318
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
@@ -340,6 +341,12 @@ def _get_english_model(
340
  ):
341
  filename = "cpu_jit-epoch-30-avg-4.pt"
342
 
 
 
 
 
 
 
343
  nn_model = _get_nn_model_filename(
344
  repo_id=repo_id,
345
  filename=filename,
@@ -824,6 +831,7 @@ chinese_models = {
824
  english_models = {
825
  "wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
826
  "yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04": _get_english_model, # noqa
 
827
  "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
828
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
829
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
 
313
  assert repo_id in [
314
  "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
315
  "yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04", # noqa
316
+ "yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19", # noqa
317
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
318
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
319
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
 
341
  ):
342
  filename = "cpu_jit-epoch-30-avg-4.pt"
343
 
344
+ if (
345
+ repo_id
346
+ == "yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19"
347
+ ):
348
+ filename = "cpu_jit-epoch-20-avg-5.pt"
349
+
350
  nn_model = _get_nn_model_filename(
351
  repo_id=repo_id,
352
  filename=filename,
 
831
  english_models = {
832
  "wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
833
  "yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04": _get_english_model, # noqa
834
+ "yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19": _get_english_model, # noqa
835
  "WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
836
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
837
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa