HoneyTian commited on
Commit
8274008
1 Parent(s): 5b60539
Files changed (1) hide show
  1. toolbox/k2_sherpa/nn_models.py +9 -1
toolbox/k2_sherpa/nn_models.py CHANGED
@@ -213,7 +213,15 @@ model_map = {
213
  "nn_model_file": "cpu_jit-epoch-30-avg-4.pt",
214
  "nn_model_file_sub_folder": "exp",
215
  "tokens_file": "tokens.txt",
216
- "tokens_file_sub_folder": "lang_bpe_500",
 
 
 
 
 
 
 
 
217
  "loader": "load_sherpa_offline_recognizer",
218
  },
219
  ],
 
213
  "nn_model_file": "cpu_jit-epoch-30-avg-4.pt",
214
  "nn_model_file_sub_folder": "exp",
215
  "tokens_file": "tokens.txt",
216
+ "tokens_file_sub_folder": "data/lang_bpe_500",
217
+ "loader": "load_sherpa_offline_recognizer",
218
+ },
219
+ {
220
+ "repo_id": "yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19",
221
+ "nn_model_file": "cpu_jit-epoch-20-avg-5.pt",
222
+ "nn_model_file_sub_folder": "exp",
223
+ "tokens_file": "tokens.txt",
224
+ "tokens_file_sub_folder": "data/lang_bpe_500",
225
  "loader": "load_sherpa_offline_recognizer",
226
  },
227
  ],