keras-hub-nightly 0.19.0.dev202502230348__py3-none-any.whl → 0.19.0.dev202502250350__py3-none-any.whl

keras_hub/src/models/gemma/gemma_attention.py

@@ -128,14 +128,8 @@ class CachedGemmaAttention(keras.layers.Layer):
                     "Please set `dropout` to 0.0."
                 )
             if attention_mask is not None:
-                while len(attention_mask.shape) < 4:
-                    attention_mask = ops.expand_dims(
-                        attention_mask, axis=1
-                    )  # Add dimension for num_heads
-                if attention_mask.shape[1] != self.num_query_heads:
-                    attention_mask = ops.tile(
-                        attention_mask, [1, self.num_query_heads, 1, 1]
-                    )
+                attention_mask = ops.expand_dims(attention_mask, axis=1)
+                attention_mask = ops.cast(attention_mask, dtype="bool")
 
             attention_output = ops.dot_product_attention(
                 query=q,

keras_hub/src/models/mobilenet/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
+from keras_hub.src.models.mobilenet.mobilenet_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, MobileNetBackbone)

keras_hub/src/models/mobilenet/mobilenet_presets.py

@@ -1,10 +1,10 @@
 """MobileNet preset configurations."""
 
 backbone_presets = {
-    "mobilenetv3_small_050": {
+    "mobilenet_v3_small_050_imagenet": {
         "metadata": {
             "description": (
-                "Small Mobilenet V3 model pre-trained on the ImageNet 1k "
+                "Small MobileNet V3 model pre-trained on the ImageNet 1k "
                 "dataset at a 224x224 resolution."
             ),
             "params": 278784,

keras_hub/src/version_utils.py

@@ -1,7 +1,7 @@
 from keras_hub.src.api_export import keras_hub_export
 
 # Unique source of truth for the version number.
-__version__ = "0.19.0.dev202502230348"
+__version__ = "0.19.0.dev202502250350"
 
 
 @keras_hub_export("keras_hub.version")

{keras_hub_nightly-0.19.0.dev202502230348.dist-info → keras_hub_nightly-0.19.0.dev202502250350.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: keras-hub-nightly
-Version: 0.19.0.dev202502230348
+Version: 0.19.0.dev202502250350
 Summary: Industry-strength Natural Language Processing extensions for Keras.
 Home-page: https://github.com/keras-team/keras-hub
 Author: Keras team

{keras_hub_nightly-0.19.0.dev202502230348.dist-info → keras_hub_nightly-0.19.0.dev202502250350.dist-info}/RECORD

@@ -8,7 +8,7 @@ keras_hub/api/tokenizers/__init__.py,sha256=mtJgQy1spfQnPAkeLoeinsT_W9iCWHlJXwzc
 keras_hub/api/utils/__init__.py,sha256=Gp1E6gG-RtKQS3PBEQEOz9PQvXkXaJ0ySGMqZ7myN7A,215
 keras_hub/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/api_export.py,sha256=9pQZK27JObxWZ96QPLBp1OBsjWigh1iuV6RglPGMRk0,1499
-keras_hub/src/version_utils.py,sha256=ETmWwXsqQMO1qU0MtwwfUMtSQplBYGumiS8GDw_fwqE,222
+keras_hub/src/version_utils.py,sha256=7049YctFzBlb70HJ2pSyLoUXeg5gpAj18FJn_dg7dEM,222
 keras_hub/src/layers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/layers/modeling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/layers/modeling/alibi_bias.py,sha256=1XBTHI52L_iJDhN_w5ydu_iMhCuTgQAxEPwcLA6BPuk,4411
@@ -183,7 +183,7 @@ keras_hub/src/models/flux/flux_presets.py,sha256=z7C_FbI1_F5YETXuWpc7Yh_0w-5N0eB
 keras_hub/src/models/flux/flux_text_to_image.py,sha256=Rf5dD2EhG0bE8Gyg9sqaA8YEexS1kdraofIkxiZDjvc,4166
 keras_hub/src/models/flux/flux_text_to_image_preprocessor.py,sha256=Fs9jr97QtmRUbRRz1kITpkuhDM2GoV3n0XSFC-qQA14,2252
 keras_hub/src/models/gemma/__init__.py,sha256=rVzOJMJ39bgVlT8UdC0t8PlN2c237GKTBmfHIsbPuOQ,251
-keras_hub/src/models/gemma/gemma_attention.py,sha256=FyCwSFckvOTg-KHKulollrsZthdw1IX3WNe5X3qq1-w,9562
+keras_hub/src/models/gemma/gemma_attention.py,sha256=wlpLfr74MOtJ18JFku6jpV2c_iRav5Q1bZiFm9e1nPU,9291
 keras_hub/src/models/gemma/gemma_backbone.py,sha256=GzAUSArw_pN9dtWQzTVhWDbW-XyWt4GyMcFLn9hwmh0,13391
 keras_hub/src/models/gemma/gemma_causal_lm.py,sha256=3OXaIXlrKqMIuUnBk-bUz-0SYFL-XkkQTWm8qRY2YII,16770
 keras_hub/src/models/gemma/gemma_causal_lm_preprocessor.py,sha256=bpKkEurWIfa6Kp9s4pz84-sBDSA6ZFNHP8nXG1fFQrg,2912
@@ -236,12 +236,12 @@ keras_hub/src/models/mit/mit_image_classifier_preprocessor.py,sha256=oNYs-pUK8Vn
 keras_hub/src/models/mit/mit_image_converter.py,sha256=Mw7nV-OzyBveGuZUNFsPPKyq9jXJVW2_cVH024CNkXM,311
 keras_hub/src/models/mit/mit_layers.py,sha256=HUJO5uhJ6jgwANpwbQdPlEVwLRVb3BZQ-Ftjg3B9XvY,9734
 keras_hub/src/models/mit/mit_presets.py,sha256=ooLrh2OoGZKxnCGnhB6BynYJtVCXH7nDDFhgQRWt36U,4528
-keras_hub/src/models/mobilenet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+keras_hub/src/models/mobilenet/__init__.py,sha256=hxkNGGj_iAMu62iooUDEPA818sNOIgjG7pXMLEMOsAE,275
 keras_hub/src/models/mobilenet/mobilenet_backbone.py,sha256=kUEDzML7MzXjUVKDY0BFn-sjGFJbu8IB8DBvG8t2nLA,28880
 keras_hub/src/models/mobilenet/mobilenet_image_classifier.py,sha256=jgyEn_WO5zLqC9UIuRV564rIY_8bMsmuUBaJA9ag5N0,2310
 keras_hub/src/models/mobilenet/mobilenet_image_classifier_preprocessor.py,sha256=yqM4wQ3ae7wXTBO0aMuvJx6XqllA7Psqzjvpm2NABXM,573
 keras_hub/src/models/mobilenet/mobilenet_image_converter.py,sha256=a3Ka0UYYK5wHSOjf2oMHSgofRazTAeUfttklVefq14w,360
-keras_hub/src/models/mobilenet/mobilenet_presets.py,sha256=TzM3geY5FJ8Myj7V_EseRgp2RvgL66pmXAUhn5qIwGw,464
+keras_hub/src/models/mobilenet/mobilenet_presets.py,sha256=PVakQu7HFRqtQdDHUzuvx_tyTWPc7BwKmWTPiHzP9C8,474
 keras_hub/src/models/mobilenet/util.py,sha256=S7j4UacmVIJ3fU8cymyAoK49eHcpWIKTOyUQiEjcbzQ,721
 keras_hub/src/models/opt/__init__.py,sha256=6Ybj8etxNaPsVcuZvaeHnKB3As92Px--dbiFAqOCIT0,239
 keras_hub/src/models/opt/opt_backbone.py,sha256=mK5z_E5mSiIX5s0w4hr4IVQpT7K46W2ajZBmuMjxwaY,5873
@@ -413,7 +413,7 @@ keras_hub/src/utils/transformers/convert_pali_gemma.py,sha256=B1leeDw96Yvu81hYum
 keras_hub/src/utils/transformers/convert_vit.py,sha256=9SUZ9utNJhW_5cj3acMn9cRy47u2eIcDsrhmzj77o9k,5187
 keras_hub/src/utils/transformers/preset_loader.py,sha256=DgGJXbTSB9Na8FIR-YWWVqQPOFxHwWrGm41EwcS_EFs,3797
 keras_hub/src/utils/transformers/safetensor_utils.py,sha256=CYUHyA4y-B61r7NDnCsFb4t_UmSwZ1k9L-8gzEd6KRg,3339
-keras_hub_nightly-0.19.0.dev202502230348.dist-info/METADATA,sha256=1EzrGRLs4Eow0j4u0bhYn95tHrP-9dVVaCvQLdCBlhE,7721
-keras_hub_nightly-0.19.0.dev202502230348.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-keras_hub_nightly-0.19.0.dev202502230348.dist-info/top_level.txt,sha256=N4J6piIWBKa38A4uV-CnIopnOEf8mHAbkNXafXm_CuA,10
-keras_hub_nightly-0.19.0.dev202502230348.dist-info/RECORD,,
+keras_hub_nightly-0.19.0.dev202502250350.dist-info/METADATA,sha256=Qd0QRQ1_wbMIdgMm4eThLxt-ujqKoUi7CkNkWC5KOJk,7721
+keras_hub_nightly-0.19.0.dev202502250350.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+keras_hub_nightly-0.19.0.dev202502250350.dist-info/top_level.txt,sha256=N4J6piIWBKa38A4uV-CnIopnOEf8mHAbkNXafXm_CuA,10
+keras_hub_nightly-0.19.0.dev202502250350.dist-info/RECORD,,