fix default attention size (#7141) (#7143)

Signed-off-by: jubick1337 <[email protected]>
NVIDIA · Aug 8, 2023 · 07beea0 · 07beea0
1 parent 9643081
commit 07beea0
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
@@ -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,