Spaces:
Build error
Build error
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
|