csukuangfj commited on
Commit
5681f66
1 Parent(s): 451aa75

update Tibetan model

Browse files
Files changed (1) hide show
  1. model.py +12 -4
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
- nn_model_filename = _get_nn_model_filename(
252
- repo_id=repo_id,
253
- filename="cpu_jit-epoch-28-avg-23-torch-1.10.0.pt",
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 = {