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

Signed-off-by: dorotat <[email protected]>
dorotat-nv · Aug 24, 2023 · 25517ca · 25517ca
1 parent e05a0c6
commit 25517ca
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,