HoneyTian commited on
Commit
03c0c2e
·
1 Parent(s): 797e507
Files changed (1) hide show
  1. toolbox/k2_sherpa/nn_models.py +24 -0
toolbox/k2_sherpa/nn_models.py CHANGED
@@ -288,6 +288,30 @@ model_map = {
288
  "tokens_file_sub_folder": "data/lang_bpe_500",
289
  "loader": "load_sherpa_offline_recognizer",
290
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
291
  ],
292
  "Chinese+English": [
293
  {
 
288
  "tokens_file_sub_folder": "data/lang_bpe_500",
289
  "loader": "load_sherpa_offline_recognizer",
290
  },
291
+ {
292
+ "repo_id": "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
293
+ "nn_model_file": "cpu_jit.pt",
294
+ "nn_model_file_sub_folder": "exp",
295
+ "tokens_file": "tokens.txt",
296
+ "tokens_file_sub_folder": "data/lang_bpe",
297
+ "loader": "load_sherpa_offline_recognizer",
298
+ },
299
+ {
300
+ "repo_id": "pkufool/icefall_asr_librispeech_conformer_ctc",
301
+ "nn_model_file": "cpu_jit.pt",
302
+ "nn_model_file_sub_folder": "exp",
303
+ "tokens_file": "tokens.txt",
304
+ "tokens_file_sub_folder": "data/lang_bpe",
305
+ "loader": "load_sherpa_offline_recognizer",
306
+ },
307
+ {
308
+ "repo_id": "WayneWiser/icefall-asr-librispeech-conformer-ctc2-jit-bpe-500-2022-07-21",
309
+ "nn_model_file": "cpu_jit.pt",
310
+ "nn_model_file_sub_folder": "exp",
311
+ "tokens_file": "tokens.txt",
312
+ "tokens_file_sub_folder": "data/lang_bpe_500",
313
+ "loader": "load_sherpa_offline_recognizer",
314
+ },
315
  ],
316
  "Chinese+English": [
317
  {