fixed masking

Author: dhantule

examples/generative/text_generation_fnet.py

@@ -262,9 +262,8 @@ def call(self, inputs, encoder_outputs, mask=None):
 
         if mask is not None:
             padding_mask = ops.cast(mask[:, None, :], "int32")
-            padding_mask = ops.minimum(padding_mask, causal_mask)
         else:
-            padding_mask = causal_mask
+            padding_mask = None
 
         attention_output_1 = self.attention_1(
             query=inputs, value=inputs, key=inputs, attention_mask=causal_mask