Remove mask = None (#2067)

keras-team · Feb 3, 2025 · 6adc92f · 6adc92f
1 parent e625699
commit 6adc92f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/keras_hub/src/models/pali_gemma/pali_gemma_vit.py b/keras_hub/src/models/pali_gemma/pali_gemma_vit.py
@@ -204,9 +204,8 @@ def __init__(
         self.intermediate_dim = intermediate_dim
 
     def compute_attention(self, x, mask=None):
-        mask = None
         if mask is not None:
-            mask = ops.cast(mask, dtype=x.dtype) if mask is not None else None
+            mask = ops.cast(mask, dtype=x.dtype)
         return self.attn(x, attention_mask=mask)[0]
 
     def build(self, input_shape):