Spaces:
Build error
Build error
Commit
·
5681f66
1
Parent(s):
451aa75
update Tibetan model
Browse files
model.py
CHANGED
|
@@ -245,13 +245,20 @@ def _get_aidatatang_200zh_pretrained_mode(repo_id: str):
|
|
| 245 |
@lru_cache(maxsize=10)
|
| 246 |
def _get_tibetan_pre_trained_model(repo_id: str):
|
| 247 |
assert repo_id in [
|
|
|
|
| 248 |
"syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless5-2022-11-29",
|
| 249 |
], repo_id
|
| 250 |
|
| 251 |
-
|
| 252 |
-
|
| 253 |
-
|
| 254 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 255 |
bpe_model_filename = _get_bpe_model_filename(repo_id=repo_id)
|
| 256 |
|
| 257 |
return OfflineAsr(
|
|
@@ -285,6 +292,7 @@ chinese_english_mixed_models = {
|
|
| 285 |
|
| 286 |
tibetan_models = {
|
| 287 |
"syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless5-2022-11-29": _get_tibetan_pre_trained_model, # noqa
|
|
|
|
| 288 |
}
|
| 289 |
|
| 290 |
all_models = {
|
|
|
|
| 245 |
@lru_cache(maxsize=10)
|
| 246 |
def _get_tibetan_pre_trained_model(repo_id: str):
|
| 247 |
assert repo_id in [
|
| 248 |
+
"syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless7-2022-12-02",
|
| 249 |
"syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless5-2022-11-29",
|
| 250 |
], repo_id
|
| 251 |
|
| 252 |
+
filename = "cpu_jit.pt"
|
| 253 |
+
if (
|
| 254 |
+
repo_id
|
| 255 |
+
== "syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless5-2022-11-29"
|
| 256 |
+
):
|
| 257 |
+
nn_model_filename = _get_nn_model_filename(
|
| 258 |
+
repo_id=repo_id,
|
| 259 |
+
filename="cpu_jit-epoch-28-avg-23-torch-1.10.0.pt",
|
| 260 |
+
)
|
| 261 |
+
|
| 262 |
bpe_model_filename = _get_bpe_model_filename(repo_id=repo_id)
|
| 263 |
|
| 264 |
return OfflineAsr(
|
|
|
|
| 292 |
|
| 293 |
tibetan_models = {
|
| 294 |
"syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless5-2022-11-29": _get_tibetan_pre_trained_model, # noqa
|
| 295 |
+
"syzym/icefall-asr-xbmu-amdo31-pruned-transducer-stateless7-2022-12-02": _get_tibetan_pre_trained_model, # noqa
|
| 296 |
}
|
| 297 |
|
| 298 |
all_models = {
|