added docs.

Signed-off-by: Vahid <vnoroozi@nvidia.com>
NVIDIA · Nov 16, 2022 · 31d9f07 · 31d9f07
1 parent 7ab6384
commit 31d9f07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py b/nemo/collections/asr/models/hybrid_rnnt_ctc_bpe_models.py
@@ -19,11 +19,11 @@
 from pytorch_lightning import Trainer
 
 from nemo.collections.asr.metrics.wer_bpe import WERBPE, CTCBPEDecoding, CTCBPEDecodingConfig
-from nemo.collections.asr.models.rnnt_bpe_models import EncDecRNNTBPEModel
+from nemo.collections.asr.models.rnnt_bpe_models import EncDecHybridRNNTCTCModel
 from nemo.utils import logging, model_utils
 
 
-class EncDecHybridRNNTCTCBPEModel(EncDecRNNTCTCModel):
+class EncDecHybridRNNTCTCBPEModel(EncDecHybridRNNTCTCModel):
     """Base class for encoder decoder RNNT-based models with auxiliary CTC decoder/loss and subword tokenization."""
 
     def __init__(self, cfg: DictConfig, trainer: Trainer = None):