HoneyTian commited on
Commit
797e507
·
1 Parent(s): e43afda
Files changed (1) hide show
  1. toolbox/k2_sherpa/nn_models.py +32 -0
toolbox/k2_sherpa/nn_models.py CHANGED
@@ -256,6 +256,38 @@ model_map = {
256
  "tokens_file_sub_folder": "data/lang_bpe_500",
257
  "loader": "load_sherpa_offline_recognizer",
258
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
259
  ],
260
  "Chinese+English": [
261
  {
 
256
  "tokens_file_sub_folder": "data/lang_bpe_500",
257
  "loader": "load_sherpa_offline_recognizer",
258
  },
259
+ {
260
+ "repo_id": "yujinqiu/sherpa-onnx-paraformer-en-2023-10-24",
261
+ "nn_model_file": "model.int8.onnx",
262
+ "nn_model_file_sub_folder": ".",
263
+ "tokens_file": "new_tokens.txt",
264
+ "tokens_file_sub_folder": ".",
265
+ "loader": "load_sherpa_offline_recognizer_from_paraformer",
266
+ },
267
+ {
268
+ "repo_id": "Zengwei/icefall-asr-librispeech-zipformer-large-2023-05-16",
269
+ "nn_model_file": "jit_script.pt",
270
+ "nn_model_file_sub_folder": "exp",
271
+ "tokens_file": "tokens.txt",
272
+ "tokens_file_sub_folder": "data/lang_bpe_500",
273
+ "loader": "load_sherpa_offline_recognizer",
274
+ },
275
+ {
276
+ "repo_id": "Zengwei/icefall-asr-librispeech-zipformer-2023-05-15",
277
+ "nn_model_file": "jit_script.pt",
278
+ "nn_model_file_sub_folder": "exp",
279
+ "tokens_file": "tokens.txt",
280
+ "tokens_file_sub_folder": "data/lang_bpe_500",
281
+ "loader": "load_sherpa_offline_recognizer",
282
+ },
283
+ {
284
+ "repo_id": "Zengwei/icefall-asr-librispeech-zipformer-small-2023-05-16",
285
+ "nn_model_file": "jit_script.pt",
286
+ "nn_model_file_sub_folder": "exp",
287
+ "tokens_file": "tokens.txt",
288
+ "tokens_file_sub_folder": "data/lang_bpe_500",
289
+ "loader": "load_sherpa_offline_recognizer",
290
+ },
291
  ],
292
  "Chinese+English": [
293
  {