Commit
·
fc7e843
1
Parent(s):
e6d227e
add a new model (fine-tuning)
Browse files
model.py
CHANGED
@@ -313,6 +313,7 @@ def _get_english_model(
|
|
313 |
assert repo_id in [
|
314 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
315 |
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04", # noqa
|
|
|
316 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless3-2022-05-13", # noqa
|
317 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11", # noqa
|
318 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14", # noqa
|
@@ -340,6 +341,12 @@ def _get_english_model(
|
|
340 |
):
|
341 |
filename = "cpu_jit-epoch-30-avg-4.pt"
|
342 |
|
|
|
|
|
|
|
|
|
|
|
|
|
343 |
nn_model = _get_nn_model_filename(
|
344 |
repo_id=repo_id,
|
345 |
filename=filename,
|
@@ -824,6 +831,7 @@ chinese_models = {
|
|
824 |
english_models = {
|
825 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
826 |
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04": _get_english_model, # noqa
|
|
|
827 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
|
828 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless8-2022-11-14": _get_english_model, # noqa
|
829 |
"csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless7-2022-11-11": _get_english_model, # noqa
|
|
|
313 |
assert repo_id in [
|
314 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02", # noqa
|
315 |
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04", # noqa
|
316 |
+
"yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19", # noqa
|
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
|
|
|
341 |
):
|
342 |
filename = "cpu_jit-epoch-30-avg-4.pt"
|
343 |
|
344 |
+
if (
|
345 |
+
repo_id
|
346 |
+
== "yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19"
|
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,
|
|
|
831 |
english_models = {
|
832 |
"wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2": _get_gigaspeech_pre_trained_model, # noqa
|
833 |
"yfyeung/icefall-asr-multidataset-pruned_transducer_stateless7-2023-05-04": _get_english_model, # noqa
|
834 |
+
"yfyeung/icefall-asr-finetune-mux-pruned_transducer_stateless7-2023-05-19": _get_english_model, # noqa
|
835 |
"WeijiZhuang/icefall-asr-librispeech-pruned-transducer-stateless8-2022-12-02": _get_english_model, # noqa
|
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
|