Merge branch 'master' into retinanet-refactor

keras-team · Feb 3, 2025 · c1eaf64 · c1eaf64
2 parents b6b1191 + c664457
commit c1eaf64
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 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):

diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
@@ -4,8 +4,8 @@ tensorflow-text~=2.18
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.5.1+cu121
-torchvision==0.20.1+cu121
+torch==2.6.0+cpu
+torchvision==0.21.0+cpu
 
 # Jax cpu-only version.
 jax[cpu]