hynt commited on
Commit
1efb8f0
·
verified ·
1 Parent(s): d7f888b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +5 -8
model.py CHANGED
@@ -700,15 +700,12 @@ def _get_gigaspeech_pre_trained_model(
700
  decoding_method: str,
701
  num_active_paths: int,
702
  ) -> sherpa.OfflineRecognizer:
703
- assert repo_id in [
704
- "wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2",
705
- ], repo_id
706
 
707
- nn_model = _get_nn_model_filename(
708
- repo_id=repo_id,
709
- filename="cpu_jit-iter-3488000-avg-20.pt",
710
- )
711
- tokens = "./giga-tokens.txt"
712
 
713
  feat_config = sherpa.FeatureConfig()
714
  feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
 
700
  decoding_method: str,
701
  num_active_paths: int,
702
  ) -> sherpa.OfflineRecognizer:
703
+ # assert repo_id in [
704
+ # "wgb14/icefall-asr-gigaspeech-pruned-transducer-stateless2",
705
+ # ], repo_id
706
 
707
+ nn_model = "jit_script.pt"
708
+ tokens = "tokens.txt"
 
 
 
709
 
710
  feat_config = sherpa.FeatureConfig()
711
  feat_config.fbank_opts.frame_opts.samp_freq = sample_rate