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

github-actions[bot] · jubick1337 · commit 07beea08eabc · 2023-08-07T17:55:07.000-07:00
Signed-off-by: jubick1337 &lt;mattyson.so@gmail.com&gt;
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,

Original file line number	Diff line number	Diff line change
`@@ -783,7 +783,8 @@ def set_default_att_context_size(self, att_context_size):`
`783`	`783`	`logging.warning(`
`784`	`784`	`f"att_context_size={att_context_size} is not among the list of the supported look-aheads: {self.att_context_size_all}"`
`785`	`785`	`)`
`786`		`- self.att_context_size = att_context_size`
	`786`	`+ if att_context_size is not None:`
	`787`	`+ self.att_context_size = att_context_size`
`787`	`788`
`788`	`789`	`def setup_streaming_params(`
`789`	`790`	`self,`