From fd8c6a420f3eae68aa8b6e6895a20e3548f896e3 Mon Sep 17 00:00:00 2001 From: Nithin Rao Date: Wed, 11 Sep 2024 13:32:11 -0400 Subject: [PATCH] add parakeet-tdt_ctc-110m model (#10461) * add parakeet-tdt_ctc-110m model Signed-off-by: Nithin Rao Koluguri * Apply isort and black reformatting Signed-off-by: nithinraok --------- Signed-off-by: Nithin Rao Koluguri Signed-off-by: nithinraok Co-authored-by: Nithin Rao Koluguri Co-authored-by: nithinraok --- nemo/collections/asr/models/hybrid_rnnt_ctc_models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py b/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py index f161454c9bae..e927faee96d6 100644 --- a/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py +++ b/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py @@ -667,4 +667,10 @@ def list_available_models(cls) -> Optional[PretrainedModelInfo]: List of available pre-trained models. """ results = [] + model = PretrainedModelInfo( + pretrained_model_name="parakeet-tdt_ctc-110m", + description="For details on this model, please refer to https://ngc.nvidia.com/catalog/models/nvidia:nemo:parakeet-tdt_ctc-110m", + location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/parakeet-tdt_ctc-110m/versions/v1/files/parakeet-tdt_ctc-110m.nemo", + ) + results.append(model) return results