Update preset path for SD 3.5 and T5 1.1 (#1961)

* Update stable_diffusion_3_presets.py * Update t5_presets.py * Update stable_diffusion_3_presets.py * upload new preset path
keras-team · Oct 30, 2024 · 97f344a · 97f344a
1 parent bd57aed
commit 97f344a
Show file tree

Hide file tree

Showing 5 changed files with 49 additions and 29 deletions.
diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py
@@ -12,7 +12,7 @@
             "path": "efficientnet",
             "model_card": "https://arxiv.org/abs/1905.11946",
         },
-        "kaggle_handle": "kaggle://kerashub/efficientnet/keras/efficientnet_b0_ra_imagenet",
+        "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet",
     },
     "efficientnet_b1_ft_imagenet": {
         "metadata": {
@@ -24,6 +24,6 @@
             "path": "efficientnet",
             "model_card": "https://arxiv.org/abs/1905.11946",
         },
-        "kaggle_handle": "kaggle://kerashub/efficientnet/keras/efficientnet_b1_ft_imagenet",
+        "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet",
     },
 }
diff --git a/keras_hub/src/models/resnet/resnet_presets.py b/keras_hub/src/models/resnet/resnet_presets.py
@@ -91,7 +91,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_18_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_18_imagenet",
     },
     "resnet_vd_34_imagenet": {
         "metadata": {
@@ -105,7 +105,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_34_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_34_imagenet",
     },
     "resnet_vd_50_imagenet": {
         "metadata": {
@@ -119,7 +119,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_imagenet",
     },
     "resnet_vd_50_ssld_imagenet": {
         "metadata": {
@@ -133,7 +133,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_imagenet",
     },
     "resnet_vd_50_ssld_v2_imagenet": {
         "metadata": {
@@ -147,7 +147,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_v2_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_v2_imagenet",
     },
     "resnet_vd_50_ssld_v2_fix_imagenet": {
         "metadata": {
@@ -162,7 +162,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_50_ssld_v2_fix_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_v2_fix_imagenet",
     },
     "resnet_vd_101_imagenet": {
         "metadata": {
@@ -176,7 +176,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_101_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_101_imagenet",
     },
     "resnet_vd_101_ssld_imagenet": {
         "metadata": {
@@ -190,7 +190,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_101_ssld_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_101_ssld_imagenet",
     },
     "resnet_vd_152_imagenet": {
         "metadata": {
@@ -204,7 +204,7 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_152_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_152_imagenet",
     },
     "resnet_vd_200_imagenet": {
         "metadata": {
@@ -218,6 +218,6 @@
             "path": "resnet",
             "model_card": "https://arxiv.org/abs/1812.01187",
         },
-        "kaggle_handle": "kaggle://kerashub/resnetvd/keras/resnet_vd_200_imagenet",
+        "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_200_imagenet",
     },
 }
diff --git a/keras_hub/src/models/segformer/segformer_presets.py b/keras_hub/src/models/segformer/segformer_presets.py
@@ -10,7 +10,7 @@
             "official_name": "SegFormerB0",
             "path": "segformer_b0",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b0_ade20k_512",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b0_ade20k_512",
     },
     "segformer_b1_ade20k_512": {
         "metadata": {
@@ -21,7 +21,7 @@
             "official_name": "SegFormerB1",
             "path": "segformer_b1",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b1_ade20k_512",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b1_ade20k_512",
     },
     "segformer_b2_ade20k_512": {
         "metadata": {
@@ -32,7 +32,7 @@
             "official_name": "SegFormerB2",
             "path": "segformer_b2",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b2_ade20k_512",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b2_ade20k_512",
     },
     "segformer_b3_ade20k_512": {
         "metadata": {
@@ -43,7 +43,7 @@
             "official_name": "SegFormerB3",
             "path": "segformer_b3",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b3_ade20k_512",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b3_ade20k_512",
     },
     "segformer_b4_ade20k_512": {
         "metadata": {
@@ -54,7 +54,7 @@
             "official_name": "SegFormerB4",
             "path": "segformer_b4",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b4_ade20k_512",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b4_ade20k_512",
     },
     "segformer_b5_ade20k_640": {
         "metadata": {
@@ -65,7 +65,7 @@
             "official_name": "SegFormerB5",
             "path": "segformer_b5",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b5_ade20k_640",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b5_ade20k_640",
     },
     "segformer_b0_cityscapes_1024": {
         "metadata": {
@@ -76,7 +76,7 @@
             "official_name": "SegFormerB0",
             "path": "segformer_b0",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b0_cityscapes_1024",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b0_cityscapes_1024",
     },
     "segformer_b1_cityscapes_1024": {
         "metadata": {
@@ -87,7 +87,7 @@
             "official_name": "SegFormerB1",
             "path": "segformer_b1",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b1_ade20k_512",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b1_ade20k_512",
     },
     "segformer_b2_cityscapes_1024": {
         "metadata": {
@@ -98,7 +98,7 @@
             "official_name": "SegFormerB2",
             "path": "segformer_b2",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b2_cityscapes_1024",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b2_cityscapes_1024",
     },
     "segformer_b3_cityscapes_1024": {
         "metadata": {
@@ -109,7 +109,7 @@
             "official_name": "SegFormerB3",
             "path": "segformer_b3",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b3_cityscapes_1024",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b3_cityscapes_1024",
     },
     "segformer_b4_cityscapes_1024": {
         "metadata": {
@@ -120,7 +120,7 @@
             "official_name": "SegFormerB4",
             "path": "segformer_b4",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b4_cityscapes_1024",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b4_cityscapes_1024",
     },
     "segformer_b5_cityscapes_1024": {
         "metadata": {
@@ -131,6 +131,6 @@
             "official_name": "SegFormerB5",
             "path": "segformer_b5",
         },
-        "kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b5_cityscapes_1024",
+        "kaggle_handle": "kaggle://keras/segformer/keras/segformer_b5_cityscapes_1024",
     },
 }
diff --git a/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py b/keras_hub/src/models/stable_diffusion_3/stable_diffusion_3_presets.py
@@ -27,7 +27,7 @@
             "path": "stable_diffusion_3",
             "model_card": "https://huggingface.co/stabilityai/stable-diffusion-3.5-large",
         },
-        "kaggle_handle": "kaggle://kerashub/stablediffusion3/keras/stable_diffusion_3.5_large/1",
+        "kaggle_handle": "kaggle://keras/stablediffusion-3.5/keras/stable_diffusion_3.5_large/1",
     },
     "stable_diffusion_3.5_large_turbo": {
         "metadata": {
@@ -43,6 +43,6 @@
             "path": "stable_diffusion_3",
             "model_card": "https://huggingface.co/stabilityai/stable-diffusion-3.5-large-turbo",
         },
-        "kaggle_handle": "kaggle://kerashub/stablediffusion3/keras/stable_diffusion_3.5_large/1",
+        "kaggle_handle": "kaggle://keras/stablediffusion-3.5/keras/stable_diffusion_3.5_large_turbo/1",
     },
 }
diff --git a/keras_hub/src/models/t5/t5_presets.py b/keras_hub/src/models/t5/t5_presets.py
@@ -22,7 +22,7 @@
             "path": "t5",
             "model_card": "https://github.com/google-research/text-to-text-transfer-transformer/blob/main/README.md",
         },
-        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_small",
+        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_small/1",
     },
     "t5_base_multi": {
         "metadata": {
@@ -45,7 +45,7 @@
             "path": "t5",
             "model_card": "https://github.com/google-research/text-to-text-transfer-transformer/blob/main/README.md",
         },
-        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_base",
+        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_base/1",
     },
     "t5_large_multi": {
         "metadata": {
@@ -68,7 +68,27 @@
             "path": "t5",
             "model_card": "https://github.com/google-research/text-to-text-transfer-transformer/blob/main/README.md",
         },
-        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_large",
+        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_large/1",
+    },
+    "t5_1.1_xl": {
+        "metadata": {
+            "description": (""),
+            "params": 2849757184,
+            "official_name": "T5 1.1",
+            "path": "t5",
+            "model_card": "https://github.com/google-research/text-to-text-transfer-transformer/blob/main/README.md",
+        },
+        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_xl/1",
+    },
+    "t5_1.1_xxl": {
+        "metadata": {
+            "description": (""),
+            "params": 11135332352,
+            "official_name": "T5 1.1",
+            "path": "t5",
+            "model_card": "https://github.com/google-research/text-to-text-transfer-transformer/blob/main/README.md",
+        },
+        "kaggle_handle": "kaggle://keras/t5/keras/t5_1.1_xxl/1",
     },
     "flan_small_multi": {
         "metadata": {