HoneyTian commited on
Commit
10a4560
1 Parent(s): ce37351
Files changed (1) hide show
  1. toolbox/k2_sherpa/nn_models.py +24 -0
toolbox/k2_sherpa/nn_models.py CHANGED
@@ -79,6 +79,30 @@ model_map = {
79
  "tokens_file_sub_folder": "data/lang_char",
80
  "loader": "load_sherpa_offline_recognizer_from_transducer",
81
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
82
  ],
83
  "English": [
84
  {
 
79
  "tokens_file_sub_folder": "data/lang_char",
80
  "loader": "load_sherpa_offline_recognizer_from_transducer",
81
  },
82
+ {
83
+ "repo_id": "zrjin/icefall-asr-aishell-zipformer-small-2023-10-24",
84
+ "encoder_model_file": "encoder-epoch-55-avg-21.onnx",
85
+ "encoder_model_file_sub_folder": "exp",
86
+ "decoder_model_file": "decoder-epoch-55-avg-21.onnx",
87
+ "decoder_model_file_sub_folder": "exp",
88
+ "joiner_model_file": "joiner-epoch-55-avg-21.onnx",
89
+ "joiner_model_file_sub_folder": "exp",
90
+ "tokens_file": "tokens.txt",
91
+ "tokens_file_sub_folder": "data/lang_char",
92
+ "loader": "load_sherpa_offline_recognizer_from_transducer",
93
+ },
94
+ {
95
+ "repo_id": "zrjin/icefall-asr-aishell-zipformer-2023-10-24",
96
+ "encoder_model_file": "encoder-epoch-55-avg-17.onnx",
97
+ "encoder_model_file_sub_folder": "exp",
98
+ "decoder_model_file": "decoder-epoch-55-avg-17.onnx",
99
+ "decoder_model_file_sub_folder": "exp",
100
+ "joiner_model_file": "joiner-epoch-55-avg-17.onnx",
101
+ "joiner_model_file_sub_folder": "exp",
102
+ "tokens_file": "tokens.txt",
103
+ "tokens_file_sub_folder": "data/lang_char",
104
+ "loader": "load_sherpa_offline_recognizer_from_transducer",
105
+ },
106
  ],
107
  "English": [
108
  {