diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index 5d83bc92c2..e94edec360 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -116,7 +116,7 @@ "params": 10589712, "path": "efficientnet", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet/1", }, "efficientnet_em_ra2_imagenet": { "metadata": { @@ -127,7 +127,7 @@ "params": 6899496, "path": "efficientnet", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet/1", }, "efficientnet_es_ra_imagenet": { "metadata": { @@ -138,7 +138,7 @@ "params": 5438392, "path": "efficientnet", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet/1", }, "efficientnet2_rw_m_agc_imagenet": { "metadata": { @@ -151,7 +151,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/2104.00298", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet2_rw_m_agc_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet2_rw_m_agc_imagenet/1", }, "efficientnet2_rw_s_ra2_imagenet": { "metadata": { @@ -164,7 +164,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/2104.00298", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet2_rw_s_ra2_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet2_rw_s_ra2_imagenet/1", }, "efficientnet2_rw_t_ra2_imagenet": { "metadata": { @@ -177,7 +177,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/2104.00298", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet2_rw_t_ra2_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet2_rw_t_ra2_imagenet/1", }, "efficientnet_lite0_ra_imagenet": { "metadata": { @@ -188,6 +188,6 @@ "params": 4652008, "path": "efficientnet", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_lite0_ra_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_lite0_ra_imagenet/1", }, } diff --git a/keras_hub/src/models/flux/flux_presets.py b/keras_hub/src/models/flux/flux_presets.py index b0607a1261..9c948cc717 100644 --- a/keras_hub/src/models/flux/flux_presets.py +++ b/keras_hub/src/models/flux/flux_presets.py @@ -1,15 +1,3 @@ """FLUX model preset configurations.""" -presets = { - "schnell": { - "metadata": { - "description": ( - "A 12 billion parameter rectified flow transformer capable of " - "generating images from text descriptions." - ), - "params": 124439808, - "path": "flux", - }, - "kaggle_handle": "TBA", - }, -} +presets = {} diff --git a/keras_hub/src/models/mit/mit_presets.py b/keras_hub/src/models/mit/mit_presets.py index 78afe07475..bb75191f60 100644 --- a/keras_hub/src/models/mit/mit_presets.py +++ b/keras_hub/src/models/mit/mit_presets.py @@ -20,7 +20,7 @@ "params": 3321962, "path": "mit", }, - "kaggle_handle": "kaggle://keras/mit/keras/mit_b0_ade20k_512/2", + "kaggle_handle": "kaggle://keras/mit/keras/mit_b0_ade20k_512/3", }, "mit_b1_ade20k_512": { "metadata": { @@ -30,7 +30,7 @@ "params": 13156554, "path": "mit", }, - "kaggle_handle": "kaggle://keras/mit/keras/mit_b1_ade20k_512/2", + "kaggle_handle": "kaggle://keras/mit/keras/mit_b1_ade20k_512/3", }, "mit_b2_ade20k_512": { "metadata": { @@ -40,7 +40,7 @@ "params": 24201418, "path": "mit", }, - "kaggle_handle": "kaggle://keras/mit/keras/mit_b2_ade20k_512/2", + "kaggle_handle": "kaggle://keras/mit/keras/mit_b2_ade20k_512/3", }, "mit_b3_ade20k_512": { "metadata": { diff --git a/keras_hub/src/models/pali_gemma/pali_gemma_presets.py b/keras_hub/src/models/pali_gemma/pali_gemma_presets.py index ffcf3ecafd..a34d11df0a 100644 --- a/keras_hub/src/models/pali_gemma/pali_gemma_presets.py +++ b/keras_hub/src/models/pali_gemma/pali_gemma_presets.py @@ -53,7 +53,7 @@ "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_896/3", }, # PaliGemma2 - "pali_gemma2_3b_ft_docci_448": { + "pali_gemma_2_ft_docci_3b_448": { "metadata": { "description": ( "3 billion parameter, image size 448, 27-layer for " @@ -66,9 +66,9 @@ "path": "pali_gemma2", "model_card": "https://www.kaggle.com/models/google/paligemma-2", }, - "kaggle_handle": "kaggle://keras/paligemma2/keras/pali_gemma2_3b_ft_docci_448/1", + "kaggle_handle": "kaggle://keras/paligemma2/keras/pali_gemma_2_ft_docci_3b_448/1", }, - "pali_gemma2_10b_ft_docci_448": { + "pali_gemma2_ft_docci_10b_448": { "metadata": { "description": ( "10 billion parameter, 27-layer for SigLIP-So400m vision " @@ -81,7 +81,7 @@ "path": "pali_gemma2", "model_card": "https://www.kaggle.com/models/google/paligemma-2", }, - "kaggle_handle": "kaggle://keras/paligemma2/keras/pali_gemma2_10b_ft_docci_448/1", + "kaggle_handle": "kaggle://keras/paligemma2/keras/pali_gemma2_ft_docci_10b_448/1", }, "pali_gemma2_pt_3b_224": { "metadata": { diff --git a/keras_hub/src/models/phi3/phi3_presets.py b/keras_hub/src/models/phi3/phi3_presets.py index 08fe0d1542..ddb66b467b 100644 --- a/keras_hub/src/models/phi3/phi3_presets.py +++ b/keras_hub/src/models/phi3/phi3_presets.py @@ -14,7 +14,7 @@ "params": 3821079552, "path": "phi3", }, - "kaggle_handle": "kaggle://keras/phi3/keras/phi3_mini_4k_instruct_en", + "kaggle_handle": "kaggle://keras/phi3/keras/phi3_mini_4k_instruct_en/1", }, "phi3_mini_128k_instruct_en": { "metadata": { @@ -28,6 +28,6 @@ "params": 3821079552, "path": "phi3", }, - "kaggle_handle": "kaggle://keras/phi3/keras/phi3_mini_128k_instruct_en", + "kaggle_handle": "kaggle://keras/phi3/keras/phi3_mini_128k_instruct_en/1", }, } diff --git a/keras_hub/src/models/resnet/resnet_presets.py b/keras_hub/src/models/resnet/resnet_presets.py index 70dd088e84..78b98976d9 100644 --- a/keras_hub/src/models/resnet/resnet_presets.py +++ b/keras_hub/src/models/resnet/resnet_presets.py @@ -77,7 +77,7 @@ "params": 11722824, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_18_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_18_imagenet/1", }, "resnet_vd_34_imagenet": { "metadata": { @@ -89,7 +89,7 @@ "params": 21838408, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_34_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_34_imagenet/1", }, "resnet_vd_50_imagenet": { "metadata": { @@ -101,7 +101,7 @@ "params": 25629512, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_imagenet/1", }, "resnet_vd_50_ssld_imagenet": { "metadata": { @@ -113,7 +113,7 @@ "params": 25629512, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_imagenet/1", }, "resnet_vd_50_ssld_v2_imagenet": { "metadata": { @@ -125,7 +125,7 @@ "params": 25629512, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_v2_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_v2_imagenet/1", }, "resnet_vd_50_ssld_v2_fix_imagenet": { "metadata": { @@ -138,7 +138,7 @@ "params": 25629512, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_v2_fix_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_50_ssld_v2_fix_imagenet/1", }, "resnet_vd_101_imagenet": { "metadata": { @@ -150,7 +150,7 @@ "params": 44673864, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_101_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_101_imagenet/1", }, "resnet_vd_101_ssld_imagenet": { "metadata": { @@ -162,7 +162,7 @@ "params": 44673864, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_101_ssld_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_101_ssld_imagenet/1", }, "resnet_vd_152_imagenet": { "metadata": { @@ -174,7 +174,7 @@ "params": 60363592, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_152_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_152_imagenet/1", }, "resnet_vd_200_imagenet": { "metadata": { @@ -186,6 +186,6 @@ "params": 74933064, "path": "resnet", }, - "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_200_imagenet", + "kaggle_handle": "kaggle://keras/resnet_vd/keras/resnet_vd_200_imagenet/1", }, }