fix default attention size (NVIDIA#7141)

Signed-off-by: Nithin Rao Koluguri <nithinraok> Co-authored-by: Nithin Rao Koluguri <nithinraok>
zhehuaichen · Oct 4, 2023 · 959c77d · 959c77d
1 parent 33ca0d7
commit 959c77d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nemo/collections/asr/modules/conformer_encoder.py b/nemo/collections/asr/modules/conformer_encoder.py
@@ -784,7 +784,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,