diff --git a/nemo/collections/asr/modules/conformer_encoder.py b/nemo/collections/asr/modules/conformer_encoder.py index 66934928fc79..323ea3341000 100644 --- a/nemo/collections/asr/modules/conformer_encoder.py +++ b/nemo/collections/asr/modules/conformer_encoder.py @@ -783,7 +783,8 @@ def set_default_att_context_size(self, att_context_size): logging.warning( f"att_context_size={att_context_size} is not among the list of the supported look-aheads: {self.att_context_size_all}" ) - self.att_context_size = att_context_size + if att_context_size is not None: + self.att_context_size = att_context_size def setup_streaming_params( self,