keras-hub-nightly 0.23.0.dev202509140416__tar.gz → 0.23.0.dev202509160412__tar.gz

{keras_hub_nightly-0.23.0.dev202509140416 → keras_hub_nightly-0.23.0.dev202509160412}/PKG-INFO

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

keras_hub_nightly-0.23.0.dev202509160412/keras_hub/src/models/d_fine/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.d_fine.d_fine_backbone import DFineBackbone
+from keras_hub.src.models.d_fine.d_fine_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, DFineBackbone)

keras_hub_nightly-0.23.0.dev202509160412/keras_hub/src/models/d_fine/d_fine_presets.py

@@ -0,0 +1,155 @@
+# Metadata for loading pretrained model weights.
+backbone_presets = {
+    "dfine_nano_coco": {
+        "metadata": {
+            "description": (
+                "D-FINE Nano model, the smallest variant in the family, "
+                "pretrained on the COCO dataset. Ideal for applications "
+                "where computational resources are limited."
+            ),
+            "params": 3788625,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_nano_coco/1",
+    },
+    "dfine_small_coco": {
+        "metadata": {
+            "description": (
+                "D-FINE Small model pretrained on the COCO dataset. Offers a "
+                "balance between performance and computational efficiency."
+            ),
+            "params": 10329321,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_small_coco/1",
+    },
+    "dfine_medium_coco": {
+        "metadata": {
+            "description": (
+                "D-FINE Medium model pretrained on the COCO dataset. A solid "
+                "baseline with strong performance for general-purpose "
+                "object detection."
+            ),
+            "params": 19621160,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_medium_coco/1",
+    },
+    "dfine_large_coco": {
+        "metadata": {
+            "description": (
+                "D-FINE Large model pretrained on the COCO dataset. Provides "
+                "high accuracy and is suitable for more demanding tasks."
+            ),
+            "params": 31344064,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_large_coco/1",
+    },
+    "dfine_xlarge_coco": {
+        "metadata": {
+            "description": (
+                "D-FINE X-Large model, the largest COCO-pretrained variant, "
+                "designed for state-of-the-art performance where accuracy "
+                "is the top priority."
+            ),
+            "params": 62834048,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_xlarge_coco/1",
+    },
+    "dfine_small_obj365": {
+        "metadata": {
+            "description": (
+                "D-FINE Small model pretrained on the large-scale Objects365 "
+                "dataset, enhancing its ability to recognize a wider "
+                "variety of objects."
+            ),
+            "params": 10623329,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_small_obj365/1",
+    },
+    "dfine_medium_obj365": {
+        "metadata": {
+            "description": (
+                "D-FINE Medium model pretrained on the Objects365 dataset. "
+                "Benefits from a larger and more diverse pretraining corpus."
+            ),
+            "params": 19988670,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_medium_obj365/1",
+    },
+    "dfine_large_obj365": {
+        "metadata": {
+            "description": (
+                "D-FINE Large model pretrained on the Objects365 dataset for "
+                "improved generalization and performance on diverse object "
+                "categories."
+            ),
+            "params": 31858578,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_large_obj365/1",
+    },
+    "dfine_xlarge_obj365": {
+        "metadata": {
+            "description": (
+                "D-FINE X-Large model pretrained on the Objects365 dataset, "
+                "offering maximum performance by leveraging a vast number "
+                "of object categories during pretraining."
+            ),
+            "params": 63348562,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_xlarge_obj365/1",
+    },
+    "dfine_small_obj2coco": {
+        "metadata": {
+            "description": (
+                "D-FINE Small model first pretrained on Objects365 and then "
+                "fine-tuned on COCO, combining broad feature learning with "
+                "benchmark-specific adaptation."
+            ),
+            "params": 10329321,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_small_obj2coco/1",
+    },
+    "dfine_medium_obj2coco": {
+        "metadata": {
+            "description": (
+                "D-FINE Medium model using a two-stage training process: "
+                "pretraining on Objects365 followed by fine-tuning on COCO."
+            ),
+            "params": 19621160,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_medium_obj2coco/1",
+    },
+    "dfine_large_obj2coco_e25": {
+        "metadata": {
+            "description": (
+                "D-FINE Large model pretrained on Objects365 and then "
+                "fine-tuned on COCO for 25 epochs. A high-performance model "
+                "with specialized tuning."
+            ),
+            "params": 31344064,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_large_obj2coco_e25/1",
+    },
+    "dfine_xlarge_obj2coco": {
+        "metadata": {
+            "description": (
+                "D-FINE X-Large model, pretrained on Objects365 and fine-tuned "
+                "on COCO, representing the most powerful model in this "
+                "series for COCO-style tasks."
+            ),
+            "params": 62834048,
+            "path": "d_fine",
+        },
+        "kaggle_handle": "kaggle://keras/d-fine/keras/dfine_xlarge_obj2coco/1",
+    },
+}

keras_hub_nightly-0.23.0.dev202509160412/keras_hub/src/models/t5gemma/t5gemma_presets.py

@@ -0,0 +1,374 @@
+# Metadata for loading pretrained model weights.
+backbone_presets = {
+    "t5gemma_s_s_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma S/S model with a small encoder and small decoder, "
+                "adapted as a UL2 model."
+            ),
+            "params": 312517632,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_s_s_ul2/1",
+    },
+    "t5gemma_s_s_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma S/S model with a small encoder and small decoder, "
+                "adapted as a prefix language model."
+            ),
+            "params": 312517632,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_s_s_prefixlm/1",
+    },
+    "t5gemma_s_s_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma S/S model with a small encoder and small decoder, "
+                "adapted as a UL2 model and fine-tuned for instruction "
+                "following."
+            ),
+            "params": 312517632,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_s_s_ul2_it/1",
+    },
+    "t5gemma_s_s_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma S/S model with a small encoder and small decoder, "
+                "adapted as a prefix language model and fine-tuned for "
+                "instruction following."
+            ),
+            "params": 312517632,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_s_s_prefixlm_it/1",
+    },
+    "t5gemma_b_b_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma B/B model with a base encoder and base decoder, "
+                "adapted as a UL2 model."
+            ),
+            "params": 591490560,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_b_b_ul2/1",
+    },
+    "t5gemma_b_b_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma B/B model with a base encoder and base decoder, "
+                "adapted as a prefix language model."
+            ),
+            "params": 591490560,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_b_b_prefixlm/1",
+    },
+    "t5gemma_b_b_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma B/B model with a base encoder and base decoder, "
+                "adapted as a UL2 model and fine-tuned for instruction "
+                "following."
+            ),
+            "params": 591490560,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_b_b_ul2_it/1",
+    },
+    "t5gemma_b_b_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma B/B model with a base encoder and base decoder, "
+                "adapted as a prefix language model and fine-tuned for "
+                "instruction following."
+            ),
+            "params": 591490560,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_b_b_prefixlm_it/1",
+    },
+    "t5gemma_l_l_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma L/L model with a large encoder and large decoder, "
+                "adapted as a UL2 model."
+            ),
+            "params": 1241761792,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_l_l_ul2/1",
+    },
+    "t5gemma_l_l_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma L/L model with a large encoder and large decoder, "
+                "adapted as a prefix language model."
+            ),
+            "params": 1241761792,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_l_l_prefixlm/1",
+    },
+    "t5gemma_l_l_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma L/L model with a large encoder and large decoder, "
+                "adapted as a UL2 model and fine-tuned for instruction "
+                "following."
+            ),
+            "params": 1241761792,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_l_l_ul2_it/1",
+    },
+    "t5gemma_l_l_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma L/L model with a large encoder and large decoder, "
+                "adapted as a prefix language model and fine-tuned for "
+                "instruction following."
+            ),
+            "params": 1241761792,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_l_l_prefixlm_it/1",
+    },
+    "t5gemma_ml_ml_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma ML/ML model with a medium-large encoder and "
+                "medium-large decoder, adapted as a UL2 model."
+            ),
+            "params": 2200345344,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_ml_ml_ul2/1",
+    },
+    "t5gemma_ml_ml_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma ML/ML model with a medium-large encoder and "
+                "medium-large decoder, adapted as a prefix language model."
+            ),
+            "params": 2200345344,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_ml_ml_prefixlm/1",
+    },
+    "t5gemma_ml_ml_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma ML/ML model with a medium-large encoder and "
+                "medium-large decoder, adapted as a UL2 model and fine-tuned "
+                "for instruction following."
+            ),
+            "params": 2200345344,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_ml_ml_ul2_it/1",
+    },
+    "t5gemma_ml_ml_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma ML/ML model with a medium-large encoder and "
+                "medium-large decoder, adapted as a prefix language model and "
+                "fine-tuned for instruction following."
+            ),
+            "params": 2200345344,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_ml_ml_prefixlm_it/1",
+    },
+    "t5gemma_xl_xl_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma XL/XL model with an extra-large encoder and "
+                "extra-large decoder, adapted as a UL2 model."
+            ),
+            "params": 3766980608,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_xl_xl_ul2/1",
+    },
+    "t5gemma_xl_xl_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma XL/XL model with an extra-large encoder and "
+                "extra-large decoder, adapted as a prefix language model."
+            ),
+            "params": 3766980608,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_xl_xl_prefixlm/1",
+    },
+    "t5gemma_xl_xl_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma XL/XL model with an extra-large encoder and "
+                "extra-large decoder, adapted as a UL2 model and fine-tuned "
+                "for instruction following."
+            ),
+            "params": 3766980608,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_xl_xl_ul2_it/1",
+    },
+    "t5gemma_xl_xl_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma XL/XL model with an extra-large encoder and "
+                "extra-large decoder, adapted as a prefix language model and "
+                "fine-tuned for instruction following."
+            ),
+            "params": 3766980608,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_xl_xl_prefixlm_it/1",
+    },
+    "t5gemma_2b_2b_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma 2B/2B model with a 2-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a UL2 model."
+            ),
+            "params": 5596853760,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_2b_2b_ul2/1",
+    },
+    "t5gemma_2b_2b_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma 2B/2B model with a 2-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a prefix language "
+                "model."
+            ),
+            "params": 5596853760,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_2b_2b_prefixlm/1",
+    },
+    "t5gemma_2b_2b_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma 2B/2B model with a 2-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a UL2 model and "
+                "fine-tuned for instruction following."
+            ),
+            "params": 5596853760,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_2b_2b_ul2_it/1",
+    },
+    "t5gemma_2b_2b_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma 2B/2B model with a 2-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a prefix language "
+                "model and fine-tuned for instruction following."
+            ),
+            "params": 5596853760,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_2b_2b_prefixlm_it/1",
+    },
+    "t5gemma_9b_2b_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/2B model with a 9-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a UL2 model."
+            ),
+            "params": 12292375296,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_2b_ul2/1",
+    },
+    "t5gemma_9b_2b_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/2B model with a 9-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a prefix language "
+                "model."
+            ),
+            "params": 12292375296,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_2b_prefixlm/1",
+    },
+    "t5gemma_9b_2b_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/2B model with a 9-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a UL2 model and "
+                "fine-tuned for instruction following."
+            ),
+            "params": 12292375296,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_2b_ul2_it/1",
+    },
+    "t5gemma_9b_2b_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/2B model with a 9-billion-parameter encoder and "
+                "2-billion-parameter decoder, adapted as a prefix language "
+                "model and fine-tuned for instruction following."
+            ),
+            "params": 12292375296,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_2b_prefixlm_it/1",
+    },
+    "t5gemma_9b_9b_ul2": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/9B model with a 9-billion-parameter encoder and "
+                "9-billion-parameter decoder, adapted as a UL2 model."
+            ),
+            "params": 20333401088,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_9b_ul2/1",
+    },
+    "t5gemma_9b_9b_prefixlm": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/9B model with a 9-billion-parameter encoder and "
+                "9-billion-parameter decoder, adapted as a prefix language "
+                "model."
+            ),
+            "params": 20333401088,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_9b_prefixlm/1",
+    },
+    "t5gemma_9b_9b_ul2_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/9B model with a 9-billion-parameter encoder and "
+                "9-billion-parameter decoder, adapted as a UL2 model and "
+                "fine-tuned for instruction following."
+            ),
+            "params": 20333401088,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_9b_ul2_it/1",
+    },
+    "t5gemma_9b_9b_prefixlm_it": {
+        "metadata": {
+            "description": (
+                "T5Gemma 9B/9B model with a 9-billion-parameter encoder and "
+                "9-billion-parameter decoder, adapted as a prefix language "
+                "model and fine-tuned for instruction following."
+            ),
+            "params": 20333401088,
+            "path": "t5gemma",
+        },
+        "kaggle_handle": "kaggle://keras/t5-gemma/keras/t5gemma_9b_9b_prefixlm_it/1",
+    },
+}

{keras_hub_nightly-0.23.0.dev202509140416 → keras_hub_nightly-0.23.0.dev202509160412}/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.dev202509140416"
+__version__ = "0.23.0.dev202509160412"
 
 
 @keras_hub_export("keras_hub.version")

{keras_hub_nightly-0.23.0.dev202509140416 → keras_hub_nightly-0.23.0.dev202509160412}/keras_hub_nightly.egg-info/PKG-INFO

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

keras_hub_nightly-0.23.0.dev202509140416/keras_hub/src/models/d_fine/d_fine_presets.py

@@ -1,2 +0,0 @@
-# Metadata for loading pretrained model weights.
-backbone_presets = {}

keras_hub_nightly-0.23.0.dev202509140416/keras_hub/src/models/t5gemma/t5gemma_presets.py

@@ -1,15 +0,0 @@
-# Metadata for loading pretrained model weights.
-backbone_presets = {
-    "t5gemma_b_b_prefixlm_it": {
-        "metadata": {
-            "description": (
-                "T5Gemma B/B model with a base encoder and base decoder, "
-                "adapted as a prefix language model and fine-tuned for "
-                "instruction following."
-            ),
-            "params": 591490560,
-            "path": "t5gemma",
-        },
-        "kaggle_handle": "kaggle://harshaljanjani/t5gemma/keras/t5gemma_b_b_prefixlm_it",
-    },
-}