keras-hub-nightly 0.16.1.dev202410170342__py3-none-any.whl → 0.16.1.dev202410190340__py3-none-any.whl

keras_hub/api/layers/__init__.py

@@ -40,9 +40,7 @@ from keras_hub.src.models.deeplab_v3.deeplab_v3_image_converter import (
 from keras_hub.src.models.densenet.densenet_image_converter import (
     DenseNetImageConverter,
 )
-from keras_hub.src.models.mix_transformer.mix_transformer_image_converter import (
-    MiTImageConverter,
-)
+from keras_hub.src.models.mit.mit_image_converter import MiTImageConverter
 from keras_hub.src.models.pali_gemma.pali_gemma_image_converter import (
     PaliGemmaImageConverter,
 )
@@ -52,7 +50,7 @@ from keras_hub.src.models.resnet.resnet_image_converter import (
 from keras_hub.src.models.sam.sam_image_converter import SAMImageConverter
 from keras_hub.src.models.sam.sam_mask_decoder import SAMMaskDecoder
 from keras_hub.src.models.sam.sam_prompt_encoder import SAMPromptEncoder
-from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageConverter
+from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
 from keras_hub.src.models.whisper.whisper_audio_converter import (
     WhisperAudioConverter,
 )

keras_hub/api/models/__init__.py

@@ -202,13 +202,9 @@ from keras_hub.src.models.mistral.mistral_causal_lm_preprocessor import (
     MistralCausalLMPreprocessor,
 )
 from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
-from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier import (
-    MiTImageClassifier,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier_preprocessor import (
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_classifier import MiTImageClassifier
+from keras_hub.src.models.mit.mit_image_classifier_preprocessor import (
     MiTImageClassifierPreprocessor,
 )
 from keras_hub.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
@@ -299,7 +295,7 @@ from keras_hub.src.models.text_classifier_preprocessor import (
 from keras_hub.src.models.text_to_image import TextToImage
 from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
 from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageClassifier
-from keras_hub.src.models.vgg.vgg_image_classifier import (
+from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
     VGGImageClassifierPreprocessor,
 )
 from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone

keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py

@@ -1,7 +1,7 @@
 """DeepLabV3 preset configurations."""
 
 backbone_presets = {
-    "deeplabv3_plus_resnet50_pascalvoc": {
+    "deeplab_v3_plus_resnet50_pascalvoc": {
         "metadata": {
             "description": (
                 "DeepLabV3+ model with ResNet50 as image encoder and trained on "
@@ -10,9 +10,9 @@ backbone_presets = {
             ),
             "params": 39190656,
             "official_name": "DeepLabV3",
-            "path": "deeplabv3",
+            "path": "deeplab_v3",
             "model_card": "https://arxiv.org/abs/1802.02611",
         },
-        "kaggle_handle": "kaggle://keras/deeplabv3/keras/deeplab_v3_plus_resnet50_pascalvoc/3",
+        "kaggle_handle": "kaggle://keras/deeplabv3plus/keras/deeplab_v3_plus_resnet50_pascalvoc/3",
     },
 }

keras_hub/src/models/densenet/densenet_presets.py

@@ -12,7 +12,7 @@ backbone_presets = {
             "path": "densenet",
             "model_card": "https://arxiv.org/abs/1608.06993",
         },
-        "kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/1",
+        "kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/2",
     },
     "densenet_169_imagenet": {
         "metadata": {
@@ -25,7 +25,7 @@ backbone_presets = {
             "path": "densenet",
             "model_card": "https://arxiv.org/abs/1608.06993",
         },
-        "kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/1",
+        "kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/2",
     },
     "densenet_201_imagenet": {
         "metadata": {
@@ -38,6 +38,6 @@ backbone_presets = {
             "path": "densenet",
             "model_card": "https://arxiv.org/abs/1608.06993",
         },
-        "kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/1",
+        "kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/2",
     },
 }

keras_hub/src/models/mit/__init__.py

@@ -0,0 +1,6 @@
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_classifier import MiTImageClassifier
+from keras_hub.src.models.mit.mit_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, MiTBackbone)

keras_hub/src/models/{mix_transformer/mix_transformer_backbone.py → mit/mit_backbone.py}

@@ -4,12 +4,8 @@ from keras import ops
 
 from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.models.feature_pyramid_backbone import FeaturePyramidBackbone
-from keras_hub.src.models.mix_transformer.mix_transformer_layers import (
-    HierarchicalTransformerEncoder,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_layers import (
-    OverlappingPatchingAndEmbedding,
-)
+from keras_hub.src.models.mit.mit_layers import HierarchicalTransformerEncoder
+from keras_hub.src.models.mit.mit_layers import OverlappingPatchingAndEmbedding
 
 
 @keras_hub_export("keras_hub.models.MiTBackbone")
@@ -61,7 +57,7 @@ class MiTBackbone(FeaturePyramidBackbone):
         ```python
         images = np.ones(shape=(1, 96, 96, 3))
         labels = np.zeros(shape=(1, 96, 96, 1))
-        backbone = keras_hub.models.MiTBackbone.from_preset("mit_b0_imagenet")
+        backbone = keras_hub.models.MiTBackbone.from_preset("mit_b0_ade20k_512")
 
         # Evaluate model
         model(images)

keras_hub/src/models/{mix_transformer/mix_transformer_classifier.py → mit/mit_image_classifier.py}

@@ -1,9 +1,7 @@
 from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.models.image_classifier import ImageClassifier
-from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier_preprocessor import (
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_classifier_preprocessor import (
     MiTImageClassifierPreprocessor,
 )
 

keras_hub/src/models/{mix_transformer/mix_transformer_classifier_preprocessor.py → mit/mit_image_classifier_preprocessor.py}

@@ -2,12 +2,8 @@ from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.models.image_classifier_preprocessor import (
     ImageClassifierPreprocessor,
 )
-from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_image_converter import (
-    MiTImageConverter,
-)
+from keras_hub.src.models.mit.mit_backbone import MiTBackbone
+from keras_hub.src.models.mit.mit_image_converter import MiTImageConverter
 
 
 @keras_hub_export("keras_hub.models.MiTImageClassifierPreprocessor")

keras_hub/src/models/{mix_transformer/mix_transformer_image_converter.py → mit/mit_image_converter.py}

@@ -1,6 +1,6 @@
 from keras_hub.src.api_export import keras_hub_export
 from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
-from keras_hub.src.models.mix_transformer import MiTBackbone
+from keras_hub.src.models.mit import MiTBackbone
 
 
 @keras_hub_export("keras_hub.layers.MiTImageConverter")

keras_hub/src/models/mobilenet/mobilenet_backbone.py

@@ -96,7 +96,7 @@ class MobileNetBackbone(Backbone):
         stackwise_activation,
         output_num_filters,
         inverted_res_block,
-        image_shape=(224, 224, 3),
+        image_shape=(None, None, 3),
         input_activation="hard_swish",
         output_activation="hard_swish",
         depth_multiplier=1.0,

keras_hub/src/models/pali_gemma/pali_gemma_backbone.py

@@ -61,8 +61,6 @@ class PaliGemmaBackbone(Backbone):
         vit_classifier_activation: activation function. The activation that
             is used for final output classification in the vision transformer.
         vit_name: string. The name used for vision transformer layers.
-        include_rescaling: bool. If true, the image input will be rescaled from
-            the range `[0, 255]`, to the range `[0, 1]`.
         layer_norm_epsilon: float. The epsilon value user for every layer norm
             in all transformer blocks.
         dropout: float. Dropout probability for the Transformer decoder blocks.
@@ -121,7 +119,6 @@ class PaliGemmaBackbone(Backbone):
         vit_pooling=None,
         vit_classifier_activation=None,
         vit_name=None,
-        include_rescaling=True,
         layer_norm_epsilon=1e-6,
         dropout=0,
         dtype=None,
@@ -145,7 +142,6 @@ class PaliGemmaBackbone(Backbone):
         vit_intermediate_dim = vit_intermediate_dim or 4304
         self.vit_encoder = PaliGemmaVit(
             image_size=image_size,
-            include_rescaling=include_rescaling,
             patch_size=vit_patch_size,
             num_heads=vit_num_heads,
             hidden_dim=vit_hidden_dim,
@@ -215,7 +211,6 @@ class PaliGemmaBackbone(Backbone):
         # === Config ===
         self.vocabulary_size = vocabulary_size
         self.image_size = image_size
-        self.include_rescaling = include_rescaling
         self.num_layers = num_layers
         self.num_query_heads = num_query_heads
         self.num_key_value_heads = num_key_value_heads
@@ -242,7 +237,6 @@ class PaliGemmaBackbone(Backbone):
             {
                 "vocabulary_size": self.vocabulary_size,
                 "image_size": self.image_size,
-                "include_rescaling": self.include_rescaling,
                 "num_layers": self.num_layers,
                 "num_query_heads": self.num_query_heads,
                 "num_key_value_heads": self.num_key_value_heads,

keras_hub/src/models/pali_gemma/pali_gemma_presets.py

@@ -12,7 +12,7 @@ backbone_presets = {
             "path": "pali_gemma",
             "model_card": "https://www.kaggle.com/models/google/paligemma",
         },
-        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_224/2",
+        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_224/3",
     },
     "pali_gemma_3b_mix_448": {
         "metadata": {
@@ -24,7 +24,7 @@ backbone_presets = {
             "path": "pali_gemma",
             "model_card": "https://www.kaggle.com/models/google/paligemma",
         },
-        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_448/2",
+        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_448/3",
     },
     "pali_gemma_3b_224": {
         "metadata": {
@@ -36,7 +36,7 @@ backbone_presets = {
             "path": "pali_gemma",
             "model_card": "https://www.kaggle.com/models/google/paligemma",
         },
-        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_224/2",
+        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_224/3",
     },
     "pali_gemma_3b_448": {
         "metadata": {
@@ -48,7 +48,7 @@ backbone_presets = {
             "path": "pali_gemma",
             "model_card": "https://www.kaggle.com/models/google/paligemma",
         },
-        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_448/2",
+        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_448/3",
     },
     "pali_gemma_3b_896": {
         "metadata": {
@@ -60,6 +60,6 @@ backbone_presets = {
             "path": "pali_gemma",
             "model_card": "https://www.kaggle.com/models/google/paligemma",
         },
-        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_896/2",
+        "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_896/3",
     },
 }

keras_hub/src/models/pali_gemma/pali_gemma_vit.py

@@ -410,8 +410,6 @@ class PaliGemmaVit(keras.Model):
     Args:
         image_size: int. The height/width of the image. Both height and width is
             expected to be the same.
-        include_rescaling: bool. If true, the image input will be rescaled from
-            the range `[0, 255]`, to the range `[0, 1]`.
         patch_size: int. The size of each square patch in the input image.
         num_heads: int. The number of attention heads for the vision(image)
             transformer encoder.
@@ -452,7 +450,6 @@ class PaliGemmaVit(keras.Model):
         num_layers,
         intermediate_dim,
         num_classes,
-        include_rescaling=True,
         pooling=None,
         classifier_activation=None,
         dtype=None,
@@ -463,14 +460,6 @@ class PaliGemmaVit(keras.Model):
             shape=(image_size, image_size, 3), name="images"
         )
         x = image_input  # Intermediate result.
-        # TODO we have moved this rescaling to preprocessing layers for most
-        # models. We should consider removing it here, though it would break
-        # compatibility.
-        if include_rescaling:
-            rescaling = keras.layers.Rescaling(
-                scale=1.0 / 127.5, offset=-1.0, name="rescaling"
-            )
-            x = rescaling(image_input)
         x = PaliGemmaVitEncoder(
             hidden_dim=hidden_dim,
             num_layers=num_layers,
@@ -520,7 +509,6 @@ class PaliGemmaVit(keras.Model):
         self.pooling = pooling
         self.num_classes = num_classes
         self.image_size = image_size
-        self.include_rescaling = include_rescaling
         self.patch_size = patch_size
         self.classifier_activation = keras.activations.get(
             classifier_activation
@@ -549,7 +537,6 @@ class PaliGemmaVit(keras.Model):
                     self.classifier_activation
                 ),
                 "image_size": self.image_size,
-                "include_rescaling": self.include_rescaling,
                 "patch_size": self.patch_size,
             }
         )

keras_hub/src/models/preprocessor.py

@@ -71,6 +71,22 @@ class Preprocessor(PreprocessingLayer):
     def image_converter(self, value):
         self._image_converter = value
 
+    @property
+    def image_size(self):
+        """Shortcut to get/set the image size of the image converter."""
+        if self.image_converter is None:
+            return None
+        return self.image_converter.image_size
+
+    @image_size.setter
+    def image_size(self, value):
+        if self.image_converter is None:
+            raise ValueError(
+                "Cannot set `image_size` on preprocessor if `image_converter` "
+                " is `None`."
+            )
+        self.image_converter.image_size = value
+
     def get_config(self):
         config = super().get_config()
         if self.tokenizer:

keras_hub/src/models/resnet/resnet_backbone.py

@@ -68,7 +68,7 @@ class ResNetBackbone(FeaturePyramidBackbone):
     input_data = np.random.uniform(0, 1, size=(2, 224, 224, 3))
 
     # Pretrained ResNet backbone.
-    model = keras_hub.models.ResNetBackbone.from_preset("resnet50")
+    model = keras_hub.models.ResNetBackbone.from_preset("resnet_50_imagenet")
     model(input_data)
 
     # Randomly initialized ResNetV2 backbone with a custom config.

keras_hub/src/models/resnet/resnet_presets.py

@@ -12,7 +12,7 @@ backbone_presets = {
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
-        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_18_imagenet/1",
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_18_imagenet/2",
     },
     "resnet_50_imagenet": {
         "metadata": {
@@ -25,7 +25,7 @@ backbone_presets = {
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
-        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_50_imagenet/1",
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_50_imagenet/2",
     },
     "resnet_101_imagenet": {
         "metadata": {
@@ -38,7 +38,7 @@ backbone_presets = {
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
-        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_101_imagenet/1",
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_101_imagenet/2",
     },
     "resnet_152_imagenet": {
         "metadata": {
@@ -51,7 +51,7 @@ backbone_presets = {
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
-        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_152_imagenet/1",
+        "kaggle_handle": "kaggle://keras/resnetv1/keras/resnet_152_imagenet/2",
     },
     "resnet_v2_50_imagenet": {
         "metadata": {
@@ -64,7 +64,7 @@ backbone_presets = {
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
-        "kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_50_imagenet/1",
+        "kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_50_imagenet/2",
     },
     "resnet_v2_101_imagenet": {
         "metadata": {
@@ -77,6 +77,147 @@ backbone_presets = {
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
-        "kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_101_imagenet/1",
+        "kaggle_handle": "kaggle://keras/resnetv2/keras/resnet_v2_101_imagenet/2",
+    },
+    "resnet_vd_18_imagenet": {
+        "metadata": {
+            "description": (
+                "18-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 11722824,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_18_imagenet",
+    },
+    "resnet_vd_34_imagenet": {
+        "metadata": {
+            "description": (
+                "34-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 21838408,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_34_imagenet",
+    },
+    "resnet_vd_50_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_imagenet",
+    },
+    "resnet_vd_50_ssld_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_imagenet",
+    },
+    "resnet_vd_50_ssld_v2_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation and AutoAugment."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_v2_imagenet",
+    },
+    "resnet_vd_50_ssld_v2_fix_imagenet": {
+        "metadata": {
+            "description": (
+                "50-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation, AutoAugment and "
+                "additional fine-tuning of the classification head."
+            ),
+            "params": 25629512,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_v2_fix_imagenet",
+    },
+    "resnet_vd_101_imagenet": {
+        "metadata": {
+            "description": (
+                "101-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 44673864,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_101_imagenet",
+    },
+    "resnet_vd_101_ssld_imagenet": {
+        "metadata": {
+            "description": (
+                "101-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution with knowledge distillation."
+            ),
+            "params": 44673864,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_101_ssld_imagenet",
+    },
+    "resnet_vd_152_imagenet": {
+        "metadata": {
+            "description": (
+                "152-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 60363592,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_152_imagenet",
+    },
+    "resnet_vd_200_imagenet": {
+        "metadata": {
+            "description": (
+                "200-layer ResNetVD (ResNet with bag of tricks) model "
+                "pre-trained on the ImageNet 1k dataset at a 224x224 "
+                "resolution."
+            ),
+            "params": 74933064,
+            "official_name": "ResNet",
+            "path": "resnet",
+            "model_card": "https://arxiv.org/abs/1812.01187",
+        },
+        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_200_imagenet",
     },
 }

keras_hub/src/models/sam/sam_image_segmenter.py

@@ -31,7 +31,7 @@ class SAMImageSegmenter(ImageSegmenter):
 
 
     Args:
-      backbone: A `keras_hub.models.VGGBackbone` instance.
+      backbone: A `keras_hub.models.SAMBackbone` instance.
 
     Example:
     Load pretrained model using `from_preset`.

keras_hub/src/models/sam/sam_presets.py

@@ -9,7 +9,7 @@ backbone_presets = {
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/2",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/4",
     },
     "sam_large_sa1b": {
         "metadata": {
@@ -19,7 +19,7 @@ backbone_presets = {
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/3",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/4",
     },
     "sam_huge_sa1b": {
         "metadata": {
@@ -29,6 +29,6 @@ backbone_presets = {
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/3",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/4",
     },
 }

keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py

@@ -10,7 +10,7 @@ backbone_presets = {
             ),
             "params": 2987080931,
             "official_name": "StableDiffusion3",
-            "path": "stablediffusion3",
+            "path": "stable_diffusion_3",
             "model_card": "https://arxiv.org/abs/2110.00476",
         },
         "kaggle_handle": "kaggle://keras/stablediffusion3/keras/stable_diffusion_3_medium/1",

keras_hub/src/models/task.py

@@ -280,7 +280,7 @@ class Task(PipelineModel):
 
         def highlight_number(x):
             if x is None:
-                f"[color(45)]{x}[/]"
+                return f"[color(45)]{x}[/]"
             return f"[color(34)]{x:,}[/]"  # Format number with commas.
 
         def highlight_symbol(x):
@@ -339,7 +339,10 @@ class Task(PipelineModel):
                         add_layer(layer, info)
                     elif isinstance(layer, ImageConverter):
                         info = "Image size: "
-                        info += highlight_shape(layer.image_size)
+                        image_size = layer.image_size
+                        if image_size is None:
+                            image_size = (None, None)
+                        info += highlight_shape(image_size)
                         add_layer(layer, info)
                     elif isinstance(layer, AudioConverter):
                         info = "Audio shape: "

keras_hub/src/models/vae/vae_backbone.py

@@ -10,7 +10,7 @@ from keras_hub.src.utils.keras_utils import standardize_data_format
 
 
 class VAEBackbone(Backbone):
-    """VAE backbone used in latent diffusion models.
+    """Variational Autoencoder(VAE) backbone used in latent diffusion models.
 
     When encoding, this model generates mean and log variance of the input
     images. When decoding, it reconstructs images from the latent space.
@@ -51,6 +51,18 @@ class VAEBackbone(Backbone):
             `"channels_last"`.
         dtype: `None` or str or `keras.mixed_precision.DTypePolicy`. The dtype
             to use for the model's computations and weights.
+
+    Example:
+    ```Python
+    backbone = VAEBackbone(
+        encoder_num_filters=[32, 32, 32, 32],
+        encoder_num_blocks=[1, 1, 1, 1],
+        decoder_num_filters=[32, 32, 32, 32],
+        decoder_num_blocks=[1, 1, 1, 1],
+    )
+    input_data = ops.ones((2, self.height, self.width, 3))
+    output = backbone(input_data)
+    ```
     """
 
     def __init__(

keras_hub/src/models/vgg/__init__.py

@@ -1 +1,5 @@
 from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
+from keras_hub.src.models.vgg.vgg_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, VGGBackbone)

keras_hub/src/models/vgg/vgg_backbone.py

@@ -20,7 +20,7 @@ class VGGBackbone(Backbone):
       stackwise_num_filters: list of ints, filter size for convolutional
             blocks per VGG block. For both VGG16 and VGG19 this is [
             64, 128, 256, 512, 512].
-      image_shape: tuple, optional shape tuple, defaults to (224, 224, 3).
+      image_shape: tuple, optional shape tuple, defaults to (None, None, 3).
 
     Examples:
     ```python

keras_hub/src/models/vgg/vgg_image_classifier.py

@@ -1,24 +1,12 @@
 import keras
 
 from keras_hub.src.api_export import keras_hub_export
-from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
 from keras_hub.src.models.image_classifier import ImageClassifier
-from keras_hub.src.models.image_classifier_preprocessor import (
-    ImageClassifierPreprocessor,
-)
 from keras_hub.src.models.task import Task
 from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
-
-
-@keras_hub_export("keras_hub.layers.VGGImageConverter")
-class VGGImageConverter(ImageConverter):
-    backbone_cls = VGGBackbone
-
-
-@keras_hub_export("keras_hub.models.VGGImageClassifierPreprocessor")
-class VGGImageClassifierPreprocessor(ImageClassifierPreprocessor):
-    backbone_cls = VGGBackbone
-    image_converter_cls = VGGImageConverter
+from keras_hub.src.models.vgg.vgg_image_classifier_preprocessor import (
+    VGGImageClassifierPreprocessor,
+)
 
 
 @keras_hub_export("keras_hub.models.VGGImageClassifier")
@@ -211,6 +199,7 @@ class VGGImageClassifier(ImageClassifier):
         self.pooling = pooling
         self.pooling_hidden_dim = pooling_hidden_dim
         self.dropout = dropout
+        self.preprocessor = preprocessor
 
     def get_config(self):
         # Backbone serialized in `super`

keras_hub/src/models/vgg/vgg_image_classifier_preprocessor.py

@@ -0,0 +1,12 @@
+from keras_hub.src.api_export import keras_hub_export
+from keras_hub.src.models.image_classifier_preprocessor import (
+    ImageClassifierPreprocessor,
+)
+from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
+from keras_hub.src.models.vgg.vgg_image_converter import VGGImageConverter
+
+
+@keras_hub_export("keras_hub.models.VGGImageClassifierPreprocessor")
+class VGGImageClassifierPreprocessor(ImageClassifierPreprocessor):
+    backbone_cls = VGGBackbone
+    image_converter_cls = VGGImageConverter

keras_hub/src/models/vgg/vgg_image_converter.py

@@ -0,0 +1,8 @@
+from keras_hub.src.api_export import keras_hub_export
+from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
+from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
+
+
+@keras_hub_export("keras_hub.layers.VGGImageConverter")
+class VGGImageConverter(ImageConverter):
+    backbone_cls = VGGBackbone

keras_hub/src/models/vit_det/vit_det_backbone.py

@@ -31,7 +31,7 @@ class ViTDetBackbone(Backbone):
         global_attention_layer_indices (list): Indexes for blocks using
             global attention.
         image_shape (tuple[int], optional): The size of the input image in
-            `(H, W, C)` format. Defaults to `(1024, 1024, 3)`.
+            `(H, W, C)` format. Defaults to `(None, None, 3)`.
         patch_size (int, optional): the patch size to be supplied to the
             Patching layer to turn input images into a flattened sequence of
             patches. Defaults to `16`.
@@ -79,7 +79,7 @@ class ViTDetBackbone(Backbone):
         intermediate_dim,
         num_heads,
         global_attention_layer_indices,
-        image_shape=(1024, 1024, 3),
+        image_shape=(None, None, 3),
         patch_size=16,
         num_output_channels=256,
         use_bias=True,

keras_hub/src/utils/pipeline_model.py

@@ -232,7 +232,7 @@ class PipelineModel(keras.Model):
     ):
         data = self.preprocess_samples(x, y, sample_weight)
         x, y, sample_weight = keras.utils.unpack_x_y_sample_weight(data)
-        x = ops.convert_to_tensor(x)
+        x = tree.map_structure(ops.convert_to_tensor, x)
         if y is not None:
             y = ops.convert_to_tensor(y)
         if sample_weight is not None:
@@ -253,7 +253,7 @@ class PipelineModel(keras.Model):
     ):
         data = self.preprocess_samples(x, y, sample_weight)
         x, y, sample_weight = keras.utils.unpack_x_y_sample_weight(data)
-        x = ops.convert_to_tensor(x)
+        x = tree.map_structure(ops.convert_to_tensor, x)
         if y is not None:
             y = ops.convert_to_tensor(y)
         if sample_weight is not None:
@@ -272,7 +272,7 @@ class PipelineModel(keras.Model):
     ):
         data = self.preprocess_samples(x)
         x, _, _ = keras.utils.unpack_x_y_sample_weight(data)
-        x = ops.convert_to_tensor(x)
+        x = tree.map_structure(ops.convert_to_tensor, x)
         return super().predict_on_batch(
             x=x,
             **kwargs,

keras_hub/src/utils/timm/preset_loader.py

@@ -53,10 +53,11 @@ class TimmPresetLoader(PresetLoader):
 
     def load_image_converter(self, cls, **kwargs):
         pretrained_cfg = self.config.get("pretrained_cfg", None)
-        if not pretrained_cfg:
+        if not pretrained_cfg or "input_size" not in pretrained_cfg:
             return None
         # This assumes the same basic setup for all timm preprocessing, We may
         # need to extend this as we cover more model types.
+        input_size = pretrained_cfg["input_size"]
         mean = pretrained_cfg["mean"]
         std = pretrained_cfg["std"]
         scale = [1.0 / 255.0 / s for s in std]
@@ -65,6 +66,7 @@ class TimmPresetLoader(PresetLoader):
         if interpolation not in ("bilinear", "nearest", "bicubic"):
             interpolation = "bilinear"  # Unsupported interpolation type.
         return cls(
+            image_size=input_size[1:],
             scale=scale,
             offset=offset,
             interpolation=interpolation,

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.16.1.dev202410170342"
+__version__ = "0.16.1.dev202410190340"
 
 
 @keras_hub_export("keras_hub.version")

{keras_hub_nightly-0.16.1.dev202410170342.dist-info → keras_hub_nightly-0.16.1.dev202410190340.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: keras-hub-nightly
-Version: 0.16.1.dev202410170342
+Version: 0.16.1.dev202410190340
 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.16.1.dev202410170342.dist-info → keras_hub_nightly-0.16.1.dev202410190340.dist-info}/RECORD

@@ -1,15 +1,15 @@
 keras_hub/__init__.py,sha256=QGdXyHgYt6cMUAP1ebxwc6oR86dE0dkMxNy2eOCQtFo,855
 keras_hub/api/__init__.py,sha256=spMxsgqzjpeuC8rY4WP-2kAZ2qwwKRSbFwddXgUjqQE,524
 keras_hub/api/bounding_box/__init__.py,sha256=T8R_X7BPm0et1xaZq8565uJmid7dylsSFSj4V-rGuFQ,1097
-keras_hub/api/layers/__init__.py,sha256=NpuVqxRCKbnvuW1QjZiSy7vHsI9Ej95ZCnNp4Lm5Hj8,2515
+keras_hub/api/layers/__init__.py,sha256=OpXnXktkkpTjlufy1u2hLPqV0cidG2B40x30jQGiy9U,2481
 keras_hub/api/metrics/__init__.py,sha256=So8Ec-lOcTzn_UUMmAdzDm8RKkPu2dbRUm2px8gpUEI,381
-keras_hub/api/models/__init__.py,sha256=9-1qQyoGODam73Vlo49EI2fzTwzwgdlCshlTMRrFEqg,14884
+keras_hub/api/models/__init__.py,sha256=5EfZDUOnHStK8UE6f6ih7cQZo2ZyFeUO15T45TC1uNA,14819
 keras_hub/api/samplers/__init__.py,sha256=n-_SEXxr2LNUzK2FqVFN7alsrkx1P_HOVTeLZKeGCdE,730
 keras_hub/api/tokenizers/__init__.py,sha256=_f-r_cyUM2fjBB7iO84ThOdqqsAxHNIewJ2EBDlM0cA,2524
 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=JKhHcqjvch67-7KPLpPGS3nhs1bP6bpaXaMsSpxp0p4,222
+keras_hub/src/version_utils.py,sha256=E7QV4TlZYLCWf3urdy8QAb07u7MCrgMVt4xHY-XOT6k,222
 keras_hub/src/bounding_box/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/bounding_box/converters.py,sha256=a5po8DBm87oz2EXfi-0uEZHCMlCJPIb4-MaZIdYx3Dg,17865
 keras_hub/src/bounding_box/formats.py,sha256=YmskOz2BOSat7NaE__J9VfpSNGPJJR0znSzA4lp8MMI,3868
@@ -61,10 +61,10 @@ keras_hub/src/models/image_to_image.py,sha256=z2TfFh9DiaEj9u6hEY8May3B0etxhptttg
 keras_hub/src/models/inpaint.py,sha256=8TTusRRS7ntPoAd0BsuhEZjedtaoljI4ZbgKQ_bnF34,20411
 keras_hub/src/models/masked_lm.py,sha256=uXO_dE_hILlOC9jNr6oK6IHi9IGUqLyNGvr6nMt8Rk0,3576
 keras_hub/src/models/masked_lm_preprocessor.py,sha256=g8vrnyYwqdnSw5xppROM1Gzo_jmMWKYZoQCsKdfrFKk,5656
-keras_hub/src/models/preprocessor.py,sha256=3CWLsMpQC77w7GzM3fU3Jf-G62ldJjufKyzPVvnGdeI,7970
+keras_hub/src/models/preprocessor.py,sha256=KqUJrF24h_6h2CnkuyneqOioCa1Sd3ZA0qzq3BdLqUA,8496
 keras_hub/src/models/seq_2_seq_lm.py,sha256=w0gX-5YZjatfvAJmFAgSHyqS_BLqc8FF8DPLGK8mrgI,1864
 keras_hub/src/models/seq_2_seq_lm_preprocessor.py,sha256=HUHRbWRG5SF1pPpotGzBhXlrMh4pLFxgAoFk05FIrB4,9687
-keras_hub/src/models/task.py,sha256=VN-CClNw3EB5Byb7HyyI3CqaS140od7-dmQInmYFSKg,14414
+keras_hub/src/models/task.py,sha256=06ISrWbn7ab-H1uszIPogpt6PuM90xiXKvwrAIEsC-o,14570
 keras_hub/src/models/text_classifier.py,sha256=VBDvQUHTpJPqKp7A4VAtm35FOmJ3yMo0DW6GdX67xG0,4159
 keras_hub/src/models/text_classifier_preprocessor.py,sha256=EoWp-GHnaLnAKTdAzDmC-soAV92ATF3QozdubdV2WXI,4722
 keras_hub/src/models/text_to_image.py,sha256=7s6rB1To46A7l9ItqRw3Pe4DGRm7YnqbHJ-RyNAlLPE,12973
@@ -122,14 +122,14 @@ keras_hub/src/models/deeplab_v3/deeplab_v3_backbone.py,sha256=WyFhuLcjFPFVuNL09b
 keras_hub/src/models/deeplab_v3/deeplab_v3_image_converter.py,sha256=mRkH3HdhpV0fCcQcVXEvIX7SNk-bAMb3SAHzgK-FD5c,371
 keras_hub/src/models/deeplab_v3/deeplab_v3_image_segmeter_preprocessor.py,sha256=hR9S6lNYamY0EBDBo3e1qTCiwtftmLXrN-UYuzfw5Io,581
 keras_hub/src/models/deeplab_v3/deeplab_v3_layers.py,sha256=qmEiolOOriLAojXB67xXW9IOo717kaCGeDVZJLaGY98,7834
-keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py,sha256=lbkP16g-w2-4RKSnISwW-RfvI6qqbE8yZzjRwgiXUIU,703
+keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py,sha256=jdSycE_H2Dm1z2WHYu0WtpEJBMiAoioHgJL1gMEGLDI,709
 keras_hub/src/models/deeplab_v3/deeplab_v3_segmenter.py,sha256=tiMDcCFANHMUx3IVtW3r1P_JTazgPPsbW4IktIytKEU,3650
 keras_hub/src/models/densenet/__init__.py,sha256=r7StyamnWeeZxOk9r4ZYNbS_YVhu9YGPyXhNxljvdPg,269
 keras_hub/src/models/densenet/densenet_backbone.py,sha256=dN9lUwKzO3E2HthNV2x54ozeBEQ0ilNs5uYHshFQpT0,6723
 keras_hub/src/models/densenet/densenet_image_classifier.py,sha256=ptuV6PwgoUpmrSPqX7-a85IpWsElwcCv_G5IVkP9E_Q,530
 keras_hub/src/models/densenet/densenet_image_classifier_preprocessor.py,sha256=xDZbTw_h6pjLDzf8QmbDyMnMsFzgh-dPX1ldg9kddhg,563
 keras_hub/src/models/densenet/densenet_image_converter.py,sha256=DoxYlJVZ9uaabFhVjWOmzvhONoc8KNcQj2vQ6Z1AUpU,354
-keras_hub/src/models/densenet/densenet_presets.py,sha256=2emOQuwcWo2i1MhvXwi081S55Z7vHkQ0r6h6Z9Yn68Q,1531
+keras_hub/src/models/densenet/densenet_presets.py,sha256=QoluKQJnV391K6hoIX5X8UquD8f647u_8Ygta-UxmwE,1531
 keras_hub/src/models/distil_bert/__init__.py,sha256=3Z0w-Mt3aOR0u9RGzjHQ7B3J3qBF2pGjupDGQ9yyzoc,303
 keras_hub/src/models/distil_bert/distil_bert_backbone.py,sha256=rnAf_GokB3wAeJwVZtgUKQO_bKJIa8RavhL_ykTJpNw,6440
 keras_hub/src/models/distil_bert/distil_bert_masked_lm.py,sha256=L0DvOl01MIwqc2f6H_E8si9qVUXPd0OKknJ5Rha33TA,4275
@@ -209,15 +209,15 @@ keras_hub/src/models/mistral/mistral_layer_norm.py,sha256=nimMZ5CTPK8v9eflfrGuzq
 keras_hub/src/models/mistral/mistral_presets.py,sha256=gucgdaFAiU-vRDS1g9zWGHjbDF_jaCiljPibCF4yVqY,1329
 keras_hub/src/models/mistral/mistral_tokenizer.py,sha256=wyzR_Y2XwrDiBV3jIeBChSPiaOkVVaxFuLxMH2F6EYA,2005
 keras_hub/src/models/mistral/mistral_transformer_decoder.py,sha256=RDIIB3FhneHZP11tNUFQT9DcWawCMnrtVxtSvtnP3ts,9542
-keras_hub/src/models/mix_transformer/__init__.py,sha256=neU-h7C0sXS6OmtS5NFJeJ1lF13OW3DaUlT6LXhl6vA,409
-keras_hub/src/models/mix_transformer/mix_transformer_backbone.py,sha256=B4hdhWHZ93lS937BGSSxovDKVXQZVuWrMbFwECFoWrg,6048
-keras_hub/src/models/mix_transformer/mix_transformer_classifier.py,sha256=pVfbbTNuiZRFElCGLyNO3gknNGnut-6-L-zAVB4Nb5w,531
-keras_hub/src/models/mix_transformer/mix_transformer_classifier_preprocessor.py,sha256=lSUuMAJiyWDVH0AVjG2y684bU3msxI3_UTa_xWyLLKQ,570
-keras_hub/src/models/mix_transformer/mix_transformer_image_converter.py,sha256=WG2LjuagCxSYXkFgqd4bHyUoMLFCzTj9QjJBoptW6WM,323
-keras_hub/src/models/mix_transformer/mix_transformer_layers.py,sha256=9AbA4kCJkjeV7fAwbRns8VGn0l1pgQ3CqFPjY-99VGA,9695
-keras_hub/src/models/mix_transformer/mix_transformer_presets.py,sha256=9bxWVOLhmBdoq2I4uZfZb7wZAB-3YjuMq0T_-JJzr2w,4960
+keras_hub/src/models/mit/__init__.py,sha256=F70_0PR_nPzPdMI8XOpXDRR_nxclGjcHv3iWSWUX3w8,316
+keras_hub/src/models/mit/mit_backbone.py,sha256=0lsWM7TkwmFE3euYbI5Xe808_ua9UDPOV4hOPlCBrOo,5984
+keras_hub/src/models/mit/mit_image_classifier.py,sha256=HKj6u6AqPbxinGYPRsz_ZdW2pEHAcFsKenrGHpRMobM,480
+keras_hub/src/models/mit/mit_image_classifier_preprocessor.py,sha256=oNYs-pUK8VnzNEPcq5beYX0qfnnlbJcxY8o5s7bVQes,504
+keras_hub/src/models/mit/mit_image_converter.py,sha256=Mw7nV-OzyBveGuZUNFsPPKyq9jXJVW2_cVH024CNkXM,311
+keras_hub/src/models/mit/mit_layers.py,sha256=9AbA4kCJkjeV7fAwbRns8VGn0l1pgQ3CqFPjY-99VGA,9695
+keras_hub/src/models/mit/mit_presets.py,sha256=9bxWVOLhmBdoq2I4uZfZb7wZAB-3YjuMq0T_-JJzr2w,4960
 keras_hub/src/models/mobilenet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-keras_hub/src/models/mobilenet/mobilenet_backbone.py,sha256=nlXdMqcj18iahy60aew4ON79EHUEuNIgvKY9dToH284,18191
+keras_hub/src/models/mobilenet/mobilenet_backbone.py,sha256=fFqEH3WTspEBYnF9LEdsX7RGHEEL3CVoVDPWG1ZnqBk,18193
 keras_hub/src/models/mobilenet/mobilenet_image_classifier.py,sha256=l5jo99I0fLlbwLub5jHw07CjC-NnmuV-ySJwXGI20Ek,351
 keras_hub/src/models/opt/__init__.py,sha256=6Ybj8etxNaPsVcuZvaeHnKB3As92Px--dbiFAqOCIT0,239
 keras_hub/src/models/opt/opt_backbone.py,sha256=mK5z_E5mSiIX5s0w4hr4IVQpT7K46W2ajZBmuMjxwaY,5873
@@ -226,14 +226,14 @@ keras_hub/src/models/opt/opt_causal_lm_preprocessor.py,sha256=xHfslVMOZlAIj2V2jI
 keras_hub/src/models/opt/opt_presets.py,sha256=J1IJ5VRcZZ6UZJSLrxpbWXw39YmbRd_WQujX1a6dxHo,2329
 keras_hub/src/models/opt/opt_tokenizer.py,sha256=oDHeed4xf07tm14hj_C78BkzMuuRwRP2cRHmqYnObrs,2557
 keras_hub/src/models/pali_gemma/__init__.py,sha256=uODWTlttOOchcTLpiYHCEWMXnDxIz8ZVIeYFQN2bd8o,288
-keras_hub/src/models/pali_gemma/pali_gemma_backbone.py,sha256=srZyBsA5tulO_Fb03g9FE-vaw2j9ftfxnAy4P8cYB6o,10916
+keras_hub/src/models/pali_gemma/pali_gemma_backbone.py,sha256=Rrl7nof_gAZL2Nge1cFymCsRdwqxQjwmEEhucGspUr0,10586
 keras_hub/src/models/pali_gemma/pali_gemma_causal_lm.py,sha256=AViEs6YltUqWnIVo7J02JkXcanBgLSdwZwF56TVr8gc,11345
 keras_hub/src/models/pali_gemma/pali_gemma_causal_lm_preprocessor.py,sha256=F57y0fZ0wYYxfGIjfrJc1W9uQpViYFx5bvFjj5CqUbI,4814
 keras_hub/src/models/pali_gemma/pali_gemma_decoder_block.py,sha256=Q_sPAULiSo_ZJeXklZjCLhvOMXk8MrPZhEXtL5yNOiI,5175
 keras_hub/src/models/pali_gemma/pali_gemma_image_converter.py,sha256=5yM_jUtrFsWIieiwfFBoP7mtPmQAwywkeLKbd7fhmzk,371
-keras_hub/src/models/pali_gemma/pali_gemma_presets.py,sha256=4D6qfWuxJtY-tyo31gxAaUlhV6wF7BhL1_FgiPmTQT0,2401
+keras_hub/src/models/pali_gemma/pali_gemma_presets.py,sha256=yLLuPwhIDE7HuMNJwLw1_yhHGz3w3mvYCxVcgAtSydc,2401
 keras_hub/src/models/pali_gemma/pali_gemma_tokenizer.py,sha256=ljTiADHo0Ok88q-jVzwJIle2C8xcxnudLTsBLzIySaM,2415
-keras_hub/src/models/pali_gemma/pali_gemma_vit.py,sha256=wP1UtW0WnlRmga-JQRxWTfAZNt_q-vaF1Qy4siJDpyY,18685
+keras_hub/src/models/pali_gemma/pali_gemma_vit.py,sha256=UpmymNkwuN9iuTV2I4M6lvHnlqpZIDmPb5pAADKs-Vg,18029
 keras_hub/src/models/phi3/__init__.py,sha256=zIbf1MU-ks91mEkjTRJAsk51N3BBnXDF2JM1vO-13PQ,245
 keras_hub/src/models/phi3/phi3_attention.py,sha256=dN8QwwTP9TxPBDv0MCvObLF3nHm1H6xbYr3T1K0nmg8,9243
 keras_hub/src/models/phi3/phi3_backbone.py,sha256=fY-OY2ZrqxDHglYjTM0OCacBdEQHwj-XNmU0MnXL7iU,8885
@@ -245,11 +245,11 @@ keras_hub/src/models/phi3/phi3_presets.py,sha256=DNyPTDA7PzFC8Ys2QmR2-mxUDa8Y8Id
 keras_hub/src/models/phi3/phi3_rotary_embedding.py,sha256=WTPCN8IKq3R7kMzsES1b8JEKV-8iNi_49WkhNTXoNUk,5012
 keras_hub/src/models/phi3/phi3_tokenizer.py,sha256=bOPH14wTVVHJHq8mgzXLjsgvKMNhfO8eayevAPpjYVA,1992
 keras_hub/src/models/resnet/__init__.py,sha256=C5UqlQ6apm8WSp1bnrxB6Bi3BGaknxRQs-r3b2wpaGA,257
-keras_hub/src/models/resnet/resnet_backbone.py,sha256=mqVdGUj8YtjZ3zIhAQXgNqu8SqiQiFlYChn0rRKF_IE,31287
+keras_hub/src/models/resnet/resnet_backbone.py,sha256=6Fno9Cf8FgNzohzwpAIhGXKTLue0aEvErg5KNMGOPnI,31297
 keras_hub/src/models/resnet/resnet_image_classifier.py,sha256=nf35EKDzvBkfhHsK-s6Ks0nbhvKO7HEOYZm94YckyWE,510
 keras_hub/src/models/resnet/resnet_image_classifier_preprocessor.py,sha256=fM7gyQ0qB-RRuI4USJkRD6q9-HVfuC71e-BLTo-UhHQ,543
 keras_hub/src/models/resnet/resnet_image_converter.py,sha256=fgTxihJznGFss-y3Z-jp0JE3X1gaaB2y-f2KMwrT8Pk,342
-keras_hub/src/models/resnet/resnet_presets.py,sha256=fqyA7rXB6IwD_x7TMq40RyArzjdDbD4jLxH5OaPjWIs,2947
+keras_hub/src/models/resnet/resnet_presets.py,sha256=kgEZQtJKnK37bSKy4Ny0GdE70P71z6OOdR_H--1pYGI,8563
 keras_hub/src/models/retinanet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/models/retinanet/anchor_generator.py,sha256=43NoI7djbRudH98hUm-9fw5OEGQNRXOUYzypIZhLYhE,6750
 keras_hub/src/models/retinanet/box_matcher.py,sha256=l820r1R-ByqiyVgmZ0YFjjz0njchDda-wItzLn1X84o,10834
@@ -267,11 +267,11 @@ keras_hub/src/models/roberta/roberta_tokenizer.py,sha256=VKPrgXVT9aMKP7et2DIWKlT
 keras_hub/src/models/sam/__init__.py,sha256=fp71Q288xeE81tIOZkkudec4Acs8v4aO5WdyzCD9x-c,239
 keras_hub/src/models/sam/sam_backbone.py,sha256=fbvtGG6du7tnkcGtEsRyT9TRwPBUJ99GBolGkWR5pkc,4351
 keras_hub/src/models/sam/sam_image_converter.py,sha256=5POp3aYFu6CK3R0NNfeUBbjhguBkincSMNvlcIJXarE,324
-keras_hub/src/models/sam/sam_image_segmenter.py,sha256=gJ-O7XaSvn9KTI-QPguhAiGfvxLUBar-KVQ-EEH5kko,7680
+keras_hub/src/models/sam/sam_image_segmenter.py,sha256=2OIowfl7cF298gIlqQlOJUsx_BcI2mCDntbXR8uohZc,7680
 keras_hub/src/models/sam/sam_image_segmenter_preprocessor.py,sha256=7slvyhGoMHmSigagqIcjDJ3gX8fUJbuMBwmozC4FlCg,849
 keras_hub/src/models/sam/sam_layers.py,sha256=SE5e6tYc-lsIVfMp9khewvuA1jY-dEHQmLT00YUok4M,13862
 keras_hub/src/models/sam/sam_mask_decoder.py,sha256=9RfjoNL7GSY6I9LZ3ulUa5cIoYSPJNP4KnHvq16lnM4,9549
-keras_hub/src/models/sam/sam_presets.py,sha256=oAv_VmRiSE4dtJRp0ue_5hP7zoXeL9ykjHwECV-dzyY,1211
+keras_hub/src/models/sam/sam_presets.py,sha256=IzpNyhxa3s5zqQFPTZOuqU75I7DU6y-TvMM27gKXLAs,1211
 keras_hub/src/models/sam/sam_prompt_encoder.py,sha256=2foB7900QbzQfZjBo335XYsdjmhOnVT8fKD1CubJNVE,11801
 keras_hub/src/models/sam/sam_transformer.py,sha256=L2bdxdc2RUF1juRZ0F0Z6r0gTva1sUwEdjItJmKKf6w,5730
 keras_hub/src/models/stable_diffusion_3/__init__.py,sha256=ZKYQuaRObyhKq8GVAHmoRvlXp6FpU8ChvutVCHyXKuc,343
@@ -280,7 +280,7 @@ keras_hub/src/models/stable_diffusion_3/mmdit.py,sha256=ByFot4_I1Z6woOBYvPcbkUtY
 keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_backbone.py,sha256=QuggvAy1yvtIXFcwyXOmE_aUdhLcCEUw4FnTuqekys0,22497
 keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_image_to_image.py,sha256=6_IXkxAv588lAKEasJrXgCjQePSXs-54XrvVIlYOT60,5483
 keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_inpaint.py,sha256=tKVAQVbKOt3lWkWsQLKN9KK3WYem0-u5fonq2uBAPrc,6367
-keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py,sha256=EP8hQHj8yqP3q26kwpNnDbrbTH7UeWL8GS5Xwo4nSCE,659
+keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py,sha256=9LcbF9Okin9ba6XJd9EQJKW5AVWdrfvW8r6F6TCx7X8,661
 keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_text_to_image.py,sha256=pQOC7xMJfJHZxZRiYFtjrbjx0GXb94cNyOr9NELoXo8,4488
 keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_text_to_image_preprocessor.py,sha256=TB0KESt5dnFYiS292PbzB0LdiH23AD6aTSTGmQEuzGM,2742
 keras_hub/src/models/stable_diffusion_3/t5_encoder.py,sha256=oV7P1uwCKdGiD93zXq7kmqX0elMZQU4UvBa8wg6P1hs,5113
@@ -293,14 +293,16 @@ keras_hub/src/models/t5/t5_presets.py,sha256=95zU4cTNEZMH2yiCLptA9zhu2D4mE1Cay18
 keras_hub/src/models/t5/t5_tokenizer.py,sha256=pLTu15JeYSpVmy-2600vBc-Mxn_uHyTKts4PI2MxxBM,2517
 keras_hub/src/models/t5/t5_transformer_layer.py,sha256=uDeP84F1x7xJxki5iKe12Zn6eWD_4yVjoFXMuod-a3A,5347
 keras_hub/src/models/vae/__init__.py,sha256=i3UaSW4IJf76O7lSPE1dyxOVjuHx8iAYKivqvUbDHOw,62
-keras_hub/src/models/vae/vae_backbone.py,sha256=aYf1sGteFJ7FyR3X8Ek6QBjAT5GjRtQTK2jXhYVJeM4,6671
+keras_hub/src/models/vae/vae_backbone.py,sha256=Yk0srJhB-zfxQeAoyZdNzvxfxPxPMVie0nqKU7cp-2M,7033
 keras_hub/src/models/vae/vae_layers.py,sha256=N83CYM1zgbl1EIjAOs3cFCkJEdxvbXkgM9ghKyljFAg,27752
-keras_hub/src/models/vgg/__init__.py,sha256=1ydFmkTOix2kOnDHie3srD4XD0dQ_7iR8OYbJzBM_YM,62
-keras_hub/src/models/vgg/vgg_backbone.py,sha256=qes1AsKwBDI7eQ3aC1uRievMkVNGXM9TNhtKLb9eZiU,3697
-keras_hub/src/models/vgg/vgg_image_classifier.py,sha256=bl6XM7l9fnOTGFreqOO3Z1jreusjhA4l7G0xjimfUKA,7829
+keras_hub/src/models/vgg/__init__.py,sha256=5ktFtITvvYja4Jg3q1LqPvGH-fMicx5wxCCpXT8aVKQ,239
+keras_hub/src/models/vgg/vgg_backbone.py,sha256=yzzindEMO1rDgf3eAv9K0rhpg4NgfGZLgxaAv5CyGM8,3699
+keras_hub/src/models/vgg/vgg_image_classifier.py,sha256=d-hlgvwbNhzR6r3q2oqEmRmuAuCpKzUwNC2JUwdzruI,7460
+keras_hub/src/models/vgg/vgg_image_classifier_preprocessor.py,sha256=M7hBbDPws5Z7oDQPigBx-upHssul7Q_p0QIv3E4yOwo,504
+keras_hub/src/models/vgg/vgg_image_converter.py,sha256=FKVrSNNBxIkiKvApzf4TZxidBb1z917Xs9nooHCcRLM,324
 keras_hub/src/models/vgg/vgg_presets.py,sha256=ltKExQdrR3E30kZPZD53tXVOsc8Gj7Krj6pzHP1UYVU,1879
 keras_hub/src/models/vit_det/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-keras_hub/src/models/vit_det/vit_det_backbone.py,sha256=GzwHXAfttExqDaGU4R2LAvng1gzjuvO3HMqUPwNUy9g,7656
+keras_hub/src/models/vit_det/vit_det_backbone.py,sha256=-ZSvfL_XSjYK_qpo5fDGY-UOI38_rqU4fD9XszK2yxU,7656
 keras_hub/src/models/vit_det/vit_layers.py,sha256=oCKeUw5ckyUAGvmFPuxIiIAqgmC3uqh85LfZcgyh964,19852
 keras_hub/src/models/whisper/__init__.py,sha256=45vTF01_e-7VzD-zvXPw1NiA9SCgDE8w0cI-6peG9cA,263
 keras_hub/src/models/whisper/whisper_audio_converter.py,sha256=aXqQ6uPI9fBSjuYbo7bMr4C0avPh3iDwrVXHEJ7W_zo,8386
@@ -345,7 +347,7 @@ keras_hub/src/tokenizers/word_piece_tokenizer.py,sha256=vP6AZgbzsRiuPCt3W_n94nsF
 keras_hub/src/tokenizers/word_piece_tokenizer_trainer.py,sha256=xUhc9EMswarzghNfrDLUFYQBExZOQxbMlfKp9G6A63k,6549
 keras_hub/src/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/utils/keras_utils.py,sha256=lrZuC8HL2lmQfbHaS_t1JUyJann_ji2iTYE0Fzos8PU,1969
-keras_hub/src/utils/pipeline_model.py,sha256=33-0vIB9KGYh2mRtyjHxBPvgGZHDusRcRy-xjki3_gg,9024
+keras_hub/src/utils/pipeline_model.py,sha256=jgzB6NQPSl0KOu08N-TazfOnXnUJbZjH2EXXhx25Ftg,9084
 keras_hub/src/utils/preset_utils.py,sha256=w45mluy4bhPPWB68waWpXFQ4MAKvSXS6llVw4rAE70s,30119
 keras_hub/src/utils/python_utils.py,sha256=N8nWeO3san4YnGkffRXG3Ix7VEIMTKSN21FX5TuL7G8,202
 keras_hub/src/utils/tensor_utils.py,sha256=JipeJUDnnvLuT-ToVQC0t9dmSzebwPG6XiZgEwGEGI4,14646
@@ -355,7 +357,7 @@ keras_hub/src/utils/timm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 keras_hub/src/utils/timm/convert_densenet.py,sha256=V-GRjWuDnlh3b1EMxqahwZ3GMwSgOa3v0HOfb2ZZ-d0,3342
 keras_hub/src/utils/timm/convert_resnet.py,sha256=ee8eTml0ffJKE8avzGoLFcpjPF63DsvoIUArAGa8Ngg,5832
 keras_hub/src/utils/timm/convert_vgg.py,sha256=MT5jGnLrzenPpe66Af_Lp1IdR9KGtsSrcmn6_UPqHvQ,2419
-keras_hub/src/utils/timm/preset_loader.py,sha256=2GJI2YeKGVovtDqc930uGta12yiyuCL9YrsTyGhqt9Y,3094
+keras_hub/src/utils/timm/preset_loader.py,sha256=PBqmnEj-fash_-GH-_ulb9YYaHAIESlOsI3wXCwKGRo,3221
 keras_hub/src/utils/transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/utils/transformers/convert_albert.py,sha256=VdKclZpCxtDWq3UbUUQZf4fR9DJK_JYZ73B4O_G9skg,7695
 keras_hub/src/utils/transformers/convert_bart.py,sha256=Tk4h9Md9rwN5wjQbGIVrC7qzDpF8kI8qm-FKL8HlUok,14411
@@ -368,7 +370,7 @@ keras_hub/src/utils/transformers/convert_mistral.py,sha256=kVhN9h1ZFVhwkNW8p3wnS
 keras_hub/src/utils/transformers/convert_pali_gemma.py,sha256=B1leeDw96Yvu81hYumf66hIid07k5NLqoeWAJgPnaLs,10649
 keras_hub/src/utils/transformers/preset_loader.py,sha256=GS44hZUuGQCtzsyn8z44ZpHdftd3DFemwV2hx2bQa-U,2738
 keras_hub/src/utils/transformers/safetensor_utils.py,sha256=rPK-Uw1CG0DX0d_UAD-r2cG9fw8GI8bvAlrcXfQ9g4c,3323
-keras_hub_nightly-0.16.1.dev202410170342.dist-info/METADATA,sha256=SEFjDxUxSNwbNsZtCnanXay7BQFJrNDM1zgriF55ATQ,7458
-keras_hub_nightly-0.16.1.dev202410170342.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-keras_hub_nightly-0.16.1.dev202410170342.dist-info/top_level.txt,sha256=N4J6piIWBKa38A4uV-CnIopnOEf8mHAbkNXafXm_CuA,10
-keras_hub_nightly-0.16.1.dev202410170342.dist-info/RECORD,,
+keras_hub_nightly-0.16.1.dev202410190340.dist-info/METADATA,sha256=MXT2uCAnfn99QJFpgrXxXlR2cvFlcPUH9_Q-SetZCpg,7458
+keras_hub_nightly-0.16.1.dev202410190340.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+keras_hub_nightly-0.16.1.dev202410190340.dist-info/top_level.txt,sha256=N4J6piIWBKa38A4uV-CnIopnOEf8mHAbkNXafXm_CuA,10
+keras_hub_nightly-0.16.1.dev202410190340.dist-info/RECORD,,

keras_hub/src/models/mix_transformer/__init__.py

@@ -1,12 +0,0 @@
-from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
-    MiTBackbone,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_classifier import (
-    MiTImageClassifier,
-)
-from keras_hub.src.models.mix_transformer.mix_transformer_presets import (
-    backbone_presets,
-)
-from keras_hub.src.utils.preset_utils import register_presets
-
-register_presets(backbone_presets, MiTBackbone)