keras-hub-nightly 0.23.0.dev202510150419__py3-none-any.whl → 0.23.0.dev202510170417__py3-none-any.whl

keras_hub/src/layers/modeling/reversible_embedding.py

@@ -245,6 +245,12 @@ class ReversibleEmbedding(keras.layers.Embedding):
                 inputs, axis=axis, to_numpy=True
             )
 
+        if mode != "int8":
+            raise NotImplementedError(
+                "Invalid quantization mode. Expected 'int8'. "
+                f"Received: quantization_mode={mode}"
+            )
+
         embeddings_shape = (self.input_dim, self.output_dim)
         if mode == "int8":
             embeddings, embeddings_scale = abs_max_quantize(

keras_hub/src/models/causal_lm.py

@@ -424,3 +424,8 @@ class CausalLM(Task):
         )
 
         export_to_safetensors(self, path)
+
+    def _post_quantize(self, mode, **kwargs):
+        super()._post_quantize(mode, **kwargs)
+        # Reset the compiled generate function.
+        self.generate_function = None

keras_hub/src/models/depth_anything/depth_anything_presets.py

@@ -1,4 +1,41 @@
 """DepthAnything model preset configurations."""
 
 # Metadata for loading pretrained model weights.
-backbone_presets = {}
+backbone_presets = {
+    "depth_anything_v2_small": {
+        "metadata": {
+            "description": (
+                "Small variant of Depth Anything V2 monocular depth estimation "
+                "(MDE) model trained on synthetic labeled images and real "
+                "unlabeled images."
+            ),
+            "params": 25_311_169,
+            "path": "depth_anything",
+        },
+        "kaggle_handle": "kaggle://keras/depth-anything/keras/depth_anything_v2_small/1",
+    },
+    "depth_anything_v2_base": {
+        "metadata": {
+            "description": (
+                "Base variant of Depth Anything V2 monocular depth estimation "
+                "(MDE) model trained on synthetic labeled images and real "
+                "unlabeled images."
+            ),
+            "params": 98_522_945,
+            "path": "depth_anything",
+        },
+        "kaggle_handle": "kaggle://keras/depth-anything/keras/depth_anything_v2_base/1",
+    },
+    "depth_anything_v2_large": {
+        "metadata": {
+            "description": (
+                "Large variant of Depth Anything V2 monocular depth estimation "
+                "(MDE) model trained on synthetic labeled images and real "
+                "unlabeled images."
+            ),
+            "params": 336_718_529,
+            "path": "depth_anything",
+        },
+        "kaggle_handle": "kaggle://keras/depth-anything/keras/depth_anything_v2_large/1",
+    },
+}

keras_hub/src/models/image_to_image.py

@@ -415,3 +415,8 @@ class ImageToImage(Task):
         # Image-to-image.
         outputs = [generate(*x) for x in inputs]
         return self._normalize_generate_outputs(outputs, input_is_scalar)
+
+    def _post_quantize(self, mode, **kwargs):
+        super()._post_quantize(mode, **kwargs)
+        # Reset the compiled generate function.
+        self.generate_function = None

keras_hub/src/models/inpaint.py

@@ -518,3 +518,8 @@ class Inpaint(Task):
         # Inpaint.
         outputs = [generate(*x) for x in inputs]
         return self._normalize_generate_outputs(outputs, input_is_scalar)
+
+    def _post_quantize(self, mode, **kwargs):
+        super()._post_quantize(mode, **kwargs)
+        # Reset the compiled generate function.
+        self.generate_function = None

keras_hub/src/models/text_to_image.py

@@ -345,3 +345,8 @@ class TextToImage(Task):
         # Text-to-image.
         outputs = [generate(x) for x in inputs]
         return self._normalize_generate_outputs(outputs, input_is_scalar)
+
+    def _post_quantize(self, mode, **kwargs):
+        super()._post_quantize(mode, **kwargs)
+        # Reset the compiled generate function.
+        self.generate_function = None

keras_hub/src/version.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.23.0.dev202510150419"
+__version__ = "0.23.0.dev202510170417"
 
 
 @keras_hub_export("keras_hub.version")

{keras_hub_nightly-0.23.0.dev202510150419.dist-info → keras_hub_nightly-0.23.0.dev202510170417.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-hub-nightly
-Version: 0.23.0.dev202510150419
+Version: 0.23.0.dev202510170417
 Summary: Pretrained models for Keras.
 Author-email: Keras team <keras-users@googlegroups.com>
 License-Expression: Apache-2.0

{keras_hub_nightly-0.23.0.dev202510150419.dist-info → keras_hub_nightly-0.23.0.dev202510170417.dist-info}/RECORD

@@ -5,7 +5,7 @@ keras_hub/models/__init__.py,sha256=E9_kQFlM75iKRiSUFEF7F8e9aqMo89-88XSSLy2sbtY,
 keras_hub/samplers/__init__.py,sha256=aFQIkiqbZpi8vjrPp2MVII4QUfE-eQjra5fMeHsoy7k,886
 keras_hub/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_hub/src/api_export.py,sha256=9pQZK27JObxWZ96QPLBp1OBsjWigh1iuV6RglPGMRk0,1499
-keras_hub/src/version.py,sha256=vcYjts3nZw0ZkBM-3v2E7wVWH_dIHZfR9h7rV9Wl2tw,222
+keras_hub/src/version.py,sha256=QhHRETV8gMH-iFFSUoHNFk78w1l9wnlAvjWBJYCdzaU,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
@@ -16,7 +16,7 @@ keras_hub/src/layers/modeling/f_net_encoder.py,sha256=zkVeO5Nk_kBZCUGq2LeDGmPEIM
 keras_hub/src/layers/modeling/masked_lm_head.py,sha256=no6XQb76KB2cUiksYC0MSdyeDOK7pn8MY6cmdCDxpKs,9015
 keras_hub/src/layers/modeling/non_max_supression.py,sha256=yAkAH1CCj_tYXgQTav39IRr_Uxn8hmzJgIxqbYQyZY8,22565
 keras_hub/src/layers/modeling/position_embedding.py,sha256=vqmmUbMU-41Ns6qwR_4N1IvVsV0arGlkiTD7D7NMS2s,4562
-keras_hub/src/layers/modeling/reversible_embedding.py,sha256=aBruxDo3nHWUpQxcS2kSLUGhV_obUPKpZKXv281AzzQ,10898
+keras_hub/src/layers/modeling/reversible_embedding.py,sha256=PVZ3G-2pIYp7fU4d6GAB9OpMUPcNGRye0hg_7XG2QBY,11096
 keras_hub/src/layers/modeling/rms_normalization.py,sha256=Ylnc9vkDw1A_ZqiKpQ09jVTAGumS5rspjdQOkH-mxf4,1084
 keras_hub/src/layers/modeling/rotary_embedding.py,sha256=uKcEyidierqdEs67QYPMQrJ1u0gxqJYT22_YGnhkQ-I,6546
 keras_hub/src/layers/modeling/sine_position_encoding.py,sha256=aLoadvQW1eeivac8gzymP740NXppblZ2C_OlErLMfN4,4063
@@ -44,7 +44,7 @@ keras_hub/src/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 keras_hub/src/models/audio_to_text.py,sha256=XoOjXtKBX6K1fz-zOXcdVo3FpjuxCMnJZh2LQcYXb_0,2726
 keras_hub/src/models/audio_to_text_preprocessor.py,sha256=GS-WWyJ6aSsPRxi_0bxvxA00h2mT2FEwSdAoQXAUYVI,3249
 keras_hub/src/models/backbone.py,sha256=BdqPsne7lIITIxn6jY6AN4vZ-Rc9VnpqTxvVNR3CS7M,12210
-keras_hub/src/models/causal_lm.py,sha256=FHGpbyFrDrnPSv5eRLfDgcpvjS6jDhSokMRl-kuumyg,18164
+keras_hub/src/models/causal_lm.py,sha256=x86PTAzoBpAdJyenPRNNBAkazUjcRLr4wb2hMs5SrQ0,18344
 keras_hub/src/models/causal_lm_preprocessor.py,sha256=nxl-sfmCfkfl6JmVRASa878QbaZUgWSA6Jdu48x4-dY,7155
 keras_hub/src/models/depth_estimator.py,sha256=JR7wtunOPrfEoDkLspoZnL2ItWhZFDeAxxw2vue5QLs,8992
 keras_hub/src/models/depth_estimator_preprocessor.py,sha256=2iE8NAUyiD2AvjZwNoXKUaOUogcE1fRzTNXLQ75GZpQ,2822
@@ -53,8 +53,8 @@ keras_hub/src/models/image_classifier.py,sha256=yt6cjhPfqs8A_eWXBsXdXFzn-aRgH2rV
 keras_hub/src/models/image_classifier_preprocessor.py,sha256=Bf7jSqHB1hX2ZWoWQS4GcXNOY_EjeoJi-_vtzCAqw4o,2690
 keras_hub/src/models/image_segmenter.py,sha256=C1bzIO59pG58iist5GLn_qnlotDpcAVxPV_8a68BkAc,2876
 keras_hub/src/models/image_segmenter_preprocessor.py,sha256=d7I2Hk0SKWyKpjRS6WYccmh_CYQBpWoj0JF5RRrU6rw,3748
-keras_hub/src/models/image_to_image.py,sha256=IJLZ6svgvcQvypwF6oe4SbJj_Zuk2-CrgHFBQcsY7n8,16753
-keras_hub/src/models/inpaint.py,sha256=fxZZrheYIK1rI6BjqZsxt9G2U0afMZR62Z87ZzuSNrQ,20815
+keras_hub/src/models/image_to_image.py,sha256=nblRd-16n5_JxKIH6IJU7bHTFRGxyCpKUilg6VjWuek,16933
+keras_hub/src/models/inpaint.py,sha256=oqdj0Q9dNG54g6sNQ5foto8saPd5Sx8kYZuHCZPBqrY,20995
 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/object_detector.py,sha256=oAK42fFBKuN0G_WM-DhygFkgQ0KsEwU_ZiU4umHywqc,3757
@@ -65,7 +65,7 @@ keras_hub/src/models/seq_2_seq_lm_preprocessor.py,sha256=DJmm4VTt8AdLtq1k9YKl_VR
 keras_hub/src/models/task.py,sha256=e9zK2zHgeOkjNACcCmAf-lGuEGF_eRoP_lKlirdIXuk,14817
 keras_hub/src/models/text_classifier.py,sha256=B6cTYDbDZW8vRvenXrLwgMMVIYMb7Pr14GvX8C_wclQ,4159
 keras_hub/src/models/text_classifier_preprocessor.py,sha256=EoWp-GHnaLnAKTdAzDmC-soAV92ATF3QozdubdV2WXI,4722
-keras_hub/src/models/text_to_image.py,sha256=NIy4S6Fh8MsbNiskAFhjmFXgRiiFqn_rOvpGOO6LlF0,13390
+keras_hub/src/models/text_to_image.py,sha256=Y2JcTBLb_l6_nnzASOXDziqP91tCPFN6m9wv6dlSe00,13570
 keras_hub/src/models/text_to_image_preprocessor.py,sha256=SKMxEABl5sy1QIA3irHTZKs7VgMdx9Cxy4IaxDU6faM,1211
 keras_hub/src/models/albert/__init__.py,sha256=rR6q_-8FujB1FXp6r4KOI7xi4gFjtAhQwXjp-MIhiyg,257
 keras_hub/src/models/albert/albert_backbone.py,sha256=4NQFo8lhv8rFiNIwQeZxxKxFwT3nKcCt36FUa6oPGok,10073
@@ -169,7 +169,7 @@ keras_hub/src/models/depth_anything/depth_anything_depth_estimator_preprocessor.
 keras_hub/src/models/depth_anything/depth_anything_image_converter.py,sha256=Xutwc8IyklFilDcc4psNBwPGRDcFlzalWXsHvEz7rUc,395
 keras_hub/src/models/depth_anything/depth_anything_layers.py,sha256=_43iEE7F8P7BL4xssjpPeFyhiDk4gGLu-wPxuqQ-nT8,27739
 keras_hub/src/models/depth_anything/depth_anything_loss.py,sha256=GJqzvLkCZrWsMDO6T2Gt_9-TYJqorfNnyOXSNgLUjQg,3389
-keras_hub/src/models/depth_anything/depth_anything_presets.py,sha256=mJ1Kg6J5lP1GkrEDH_qS1XJ__98Tjt5noRZVZaHCn9k,121
+keras_hub/src/models/depth_anything/depth_anything_presets.py,sha256=A3Afr06IRL02u-9EPNTTiGb6DgzjQavSwVmWHz0OoMc,1536
 keras_hub/src/models/depth_anything/interpolate.py,sha256=qwrPGP6wA4jZ-XcSeulhkyxPDiMRxHlC92EqSd0H5Tk,2041
 keras_hub/src/models/dinov2/__init__.py,sha256=qacZi82EfAloVND4gDLZjqgR5_yVdz_dc4mMKyCsjOA,257
 keras_hub/src/models/dinov2/dinov2_backbone.py,sha256=QH3lzE1EnxTcOSii9KS1Qx3lq0XcZMsvElB7AL_ejZY,10672
@@ -610,7 +610,7 @@ keras_hub/src/utils/transformers/export/gemma.py,sha256=xX_vfQwvFZ_-lQX4kgMNOGKL
 keras_hub/src/utils/transformers/export/hf_exporter.py,sha256=Qk52c6LIA2eMHUNY9Vy4STJSpnhLMdJ_t-3ljqhSr4k,5081
 keras_hub/tokenizers/__init__.py,sha256=kyFWYm4mb--U4xYU-2Gb1COM8xEFWNK6LcKxr8h9Ivc,4561
 keras_hub/utils/__init__.py,sha256=jXPqVGBpJr_PpYmqD8aDG-fRMlxH-ulqCR2SZMn288Y,646
-keras_hub_nightly-0.23.0.dev202510150419.dist-info/METADATA,sha256=kplsqZ2jhRdxM0KN1cXDibhuSWMWu2f3Mt41vZhyHv8,7395
-keras_hub_nightly-0.23.0.dev202510150419.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-keras_hub_nightly-0.23.0.dev202510150419.dist-info/top_level.txt,sha256=N4J6piIWBKa38A4uV-CnIopnOEf8mHAbkNXafXm_CuA,10
-keras_hub_nightly-0.23.0.dev202510150419.dist-info/RECORD,,
+keras_hub_nightly-0.23.0.dev202510170417.dist-info/METADATA,sha256=7uTd-YS4ylZmQT0k7rfgGYmhJcoHSw56H1T923OsH6I,7395
+keras_hub_nightly-0.23.0.dev202510170417.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+keras_hub_nightly-0.23.0.dev202510170417.dist-info/top_level.txt,sha256=N4J6piIWBKa38A4uV-CnIopnOEf8mHAbkNXafXm_CuA,10
+keras_hub_nightly-0.23.0.dev202510170417.dist-info/RECORD,,