Skip to content

Commit 6fa91ab

Browse files
nithinraokgenquan9
authored andcommitted
fix loading of hyb ctc rnnt bpe models when using from pretrained (NVIDIA-NeMo#15042)
* fix loading of hyb ctc rnnt bpe models when using from pretrained Signed-off-by: nithinraok <[email protected]> * Apply isort and black reformatting Signed-off-by: nithinraok <[email protected]> --------- Signed-off-by: nithinraok <[email protected]> Signed-off-by: nithinraok <[email protected]> Co-authored-by: nithinraok <[email protected]> Signed-off-by: genquan9 <[email protected]>
1 parent aa3d1cf commit 6fa91ab

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models_prompt.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
from nemo.collections.asr.parts.submodules.rnnt_decoding import RNNTBPEDecoding
3737
from nemo.collections.asr.parts.utils.rnnt_utils import Hypothesis
3838
from nemo.collections.common.data.lhotse import get_lhotse_dataloader_from_config
39-
from nemo.core.classes.common import typecheck
39+
from nemo.core.classes.common import PretrainedModelInfo, typecheck
4040
from nemo.core.classes.mixins import AccessMixin
4141
from nemo.core.neural_types import (
4242
AcousticEncodedRepresentation,
@@ -1000,3 +1000,13 @@ def setup_test_data(self, test_data_config: Optional[Union[DictConfig, Dict]]):
10001000
# preserve config
10011001
self._update_dataset_config(dataset_name='test', config=test_data_config)
10021002
self._test_dl = self._setup_dataloader_from_config(config=test_data_config)
1003+
1004+
@classmethod
1005+
def list_available_models(cls) -> List[PretrainedModelInfo]:
1006+
"""
1007+
This method returns a list of pre-trained model which can be instantiated directly from NVIDIA's NGC cloud.
1008+
1009+
Returns:
1010+
List of available pre-trained models.
1011+
"""
1012+
return None

0 commit comments

Comments
 (0)