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

add new zipformer

Browse files
Files changed (1) hide show
  1. model.py +14 -1
model.py CHANGED
@@ -26,7 +26,7 @@ os.system(
26
  "cp -v /home/user/.local/lib/python3.8/site-packages/k2/lib/*.so /home/user/.local/lib/python3.8/site-packages/sherpa/lib/"
27
  )
28
 
29
- import k2
30
  import sherpa
31
  import sherpa_onnx
32
  import numpy as np
@@ -317,6 +317,9 @@ def _get_english_model(
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
 
 
 
320
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
321
  "pkufool/icefall_asr_librispeech_conformer_ctc",
322
  "WayneWiser/icefall-asr-librispeech-conformer-ctc2-jit-bpe-500-2022-07-21",
@@ -347,6 +350,13 @@ def _get_english_model(
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,
@@ -836,6 +846,9 @@ english_models = {
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
838
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model, # noqa
 
 
 
839
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
840
  "pkufool/icefall_asr_librispeech_conformer_ctc": _get_english_model,
841
  "WayneWiser/icefall-asr-librispeech-conformer-ctc2-jit-bpe-500-2022-07-21": _get_english_model,
 
26
  "cp -v /home/user/.local/lib/python3.8/site-packages/k2/lib/*.so /home/user/.local/lib/python3.8/site-packages/sherpa/lib/"
27
  )
28
 
29
+ import k2 # noqa
30
  import sherpa
31
  import sherpa_onnx
32
  import numpy as np
 
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
320
+ "Zengwei/icefall-asr-librispeech-zipformer-large-2023-05-16", # noqa
321
+ "Zengwei/icefall-asr-librispeech-zipformer-2023-05-15", # noqa
322
+ "Zengwei/icefall-asr-librispeech-zipformer-small-2023-05-16", # noqa
323
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2",
324
  "pkufool/icefall_asr_librispeech_conformer_ctc",
325
  "WayneWiser/icefall-asr-librispeech-conformer-ctc2-jit-bpe-500-2022-07-21",
 
350
  ):
351
  filename = "cpu_jit-epoch-20-avg-5.pt"
352
 
353
+ if repo_id in (
354
+ "Zengwei/icefall-asr-librispeech-zipformer-large-2023-05-16",
355
+ "Zengwei/icefall-asr-librispeech-zipformer-2023-05-15",
356
+ "Zengwei/icefall-asr-librispeech-zipformer-small-2023-05-16",
357
+ ):
358
+ filename = "jit_script.pt"
359
+
360
  nn_model = _get_nn_model_filename(
361
  repo_id=repo_id,
362
  filename=filename,
 
846
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
847
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
848
  "csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model, # noqa
849
+ "Zengwei/icefall-asr-librispeech-zipformer-large-2023-05-16": _get_english_model, # noqa
850
+ "Zengwei/icefall-asr-librispeech-zipformer-2023-05-15": _get_english_model, # noqa
851
+ "Zengwei/icefall-asr-librispeech-zipformer-small-2023-05-16": _get_english_model, # noqa
852
  "videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
853
  "pkufool/icefall_asr_librispeech_conformer_ctc": _get_english_model,
854
  "WayneWiser/icefall-asr-librispeech-conformer-ctc2-jit-bpe-500-2022-07-21": _get_english_model,