csukuangfj
commited on
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 = {
|