From 1934506ca443ea05844b0fd5e87e434cb9f157e1 Mon Sep 17 00:00:00 2001
From: Nithin Rao Koluguri <nithinraok>
Date: Mon, 31 Jul 2023 22:55:53 -0700
Subject: [PATCH] fix default attention size

Signed-off-by: Nithin Rao Koluguri <nithinraok>
---
 nemo/collections/asr/modules/conformer_encoder.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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,