diff --git a/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py b/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py index 2b13f02ab657..11c616b1257f 100644 --- a/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py +++ b/nemo/collections/asr/models/hybrid_rnnt_ctc_models.py @@ -650,7 +650,14 @@ def list_export_subnets(self): if self.cur_decoder == 'rnnt': return ['encoder', 'decoder_joint'] else: - return ['encoder'] + return ['self'] + + @property + def output_module(self): + if self.cur_decoder == 'rnnt': + return self.decoder + else: + return self.ctc_decoder @classmethod def list_available_models(cls) -> Optional[PretrainedModelInfo]: