Update model.py
Browse files
model.py
CHANGED
|
@@ -588,6 +588,42 @@ def _get_revolab_pretrained_model(
|
|
| 588 |
|
| 589 |
return recognizer
|
| 590 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 591 |
@lru_cache(maxsize=10)
|
| 592 |
def _get_zrjin_cantonese_pre_trained_model(
|
| 593 |
repo_id: str, decoding_method: str, num_active_paths: int
|
|
|
|
| 588 |
|
| 589 |
return recognizer
|
| 590 |
|
| 591 |
+
elif repo_id == "Revolab/zipformer-scaled-up":
|
| 592 |
+
|
| 593 |
+
real_repo = 'Revolab/malaysian-pruned_transducer_stateless7'
|
| 594 |
+
|
| 595 |
+
encoder_model = _get_nn_model_filename(
|
| 596 |
+
repo_id=real_repo,
|
| 597 |
+
filename="encoder-epoch-13-avg-5.onnx",
|
| 598 |
+
subfolder="zipformer-scale-up-pretrained/export",
|
| 599 |
+
)
|
| 600 |
+
decoder_model = _get_nn_model_filename(
|
| 601 |
+
repo_id=real_repo,
|
| 602 |
+
filename="encoder-epoch-13-avg-5.onnx",
|
| 603 |
+
subfolder="zipformer-scale-up-pretrained/export",
|
| 604 |
+
)
|
| 605 |
+
|
| 606 |
+
joiner_model = _get_nn_model_filename(
|
| 607 |
+
repo_id=real_repo,
|
| 608 |
+
filename="joiner-epoch-13-avg-5.onnx",
|
| 609 |
+
subfolder="zipformer-scale-up-pretrained/export",
|
| 610 |
+
)
|
| 611 |
+
|
| 612 |
+
tokens = _get_token_filename(repo_id=real_repo, subfolder="zipformer-scale-up-pretrained/export/lang_bpe_500")
|
| 613 |
+
|
| 614 |
+
recognizer = sherpa_onnx.OfflineRecognizer.from_transducer(
|
| 615 |
+
tokens=tokens,
|
| 616 |
+
encoder=encoder_model,
|
| 617 |
+
decoder=decoder_model,
|
| 618 |
+
joiner=joiner_model,
|
| 619 |
+
num_threads=2,
|
| 620 |
+
sample_rate=16000,
|
| 621 |
+
feature_dim=80,
|
| 622 |
+
decoding_method=decoding_method,
|
| 623 |
+
)
|
| 624 |
+
|
| 625 |
+
return recognizer
|
| 626 |
+
|
| 627 |
@lru_cache(maxsize=10)
|
| 628 |
def _get_zrjin_cantonese_pre_trained_model(
|
| 629 |
repo_id: str, decoding_method: str, num_active_paths: int
|