Update model.py
Browse files
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 |
-
|
| 705 |
-
], repo_id
|
| 706 |
|
| 707 |
-
nn_model =
|
| 708 |
-
|
| 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
|