Commit
·
9fa23bb
1
Parent(s):
4abdf19
add new CTC model
Browse files
model.py
CHANGED
@@ -187,6 +187,7 @@ def _get_english_model(
|
|
187 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
188 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
189 |
"videodanchik/icefall-asr-tedlium3-conformer-ctc2",
|
|
|
190 |
], repo_id
|
191 |
|
192 |
filename = "cpu_jit.pt"
|
@@ -208,7 +209,10 @@ def _get_english_model(
|
|
208 |
)
|
209 |
subfolder = "data/lang_bpe_500"
|
210 |
|
211 |
-
if repo_id
|
|
|
|
|
|
|
212 |
subfolder = "data/lang_bpe"
|
213 |
|
214 |
tokens = _get_token_filename(repo_id=repo_id, subfolder=subfolder)
|
@@ -559,6 +563,7 @@ english_models = {
|
|
559 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|
560 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model, # noqa
|
561 |
"videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
|
|
|
562 |
"csukuangfj/wenet-english-model": _get_wenet_model,
|
563 |
}
|
564 |
|
|
|
187 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
188 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
189 |
"videodanchik/icefall-asr-tedlium3-conformer-ctc2",
|
190 |
+
"pkufool/icefall_asr_librispeech_conformer_ctc",
|
191 |
], repo_id
|
192 |
|
193 |
filename = "cpu_jit.pt"
|
|
|
209 |
)
|
210 |
subfolder = "data/lang_bpe_500"
|
211 |
|
212 |
+
if repo_id in (
|
213 |
+
"videodanchik/icefall-asr-tedlium3-conformer-ctc2",
|
214 |
+
"pkufool/icefall_asr_librispeech_conformer_ctc",
|
215 |
+
):
|
216 |
subfolder = "data/lang_bpe"
|
217 |
|
218 |
tokens = _get_token_filename(repo_id=repo_id, subfolder=subfolder)
|
|
|
563 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|
564 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13": _get_english_model, # noqa
|
565 |
"videodanchik/icefall-asr-tedlium3-conformer-ctc2": _get_english_model,
|
566 |
+
"pkufool/icefall_asr_librispeech_conformer_ctc": _get_english_model,
|
567 |
"csukuangfj/wenet-english-model": _get_wenet_model,
|
568 |
}
|
569 |
|