Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Sync HF <> Kaggle #1986

Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -165,4 +165,4 @@ Thank you to all of our wonderful contributors!

<a href="https://github.com/keras-team/keras-hub/graphs/contributors">
<img src="https://contrib.rocks/image?repo=keras-team/keras-hub" />
</a>
</a>
16 changes: 8 additions & 8 deletions keras_hub/src/models/clip/clip_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
"path": "clip",
"model_card": "https://github.com/openai/CLIP/blob/main/model-card.md",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_base_patch16/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch16/1",
},
"clip_vit_base_patch32": {
"metadata": {
Expand All @@ -26,7 +26,7 @@
"path": "clip",
"model_card": "https://github.com/openai/CLIP/blob/main/model-card.md",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_base_patch32/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch32/1",
},
"clip_vit_large_patch14": {
"metadata": {
Expand All @@ -39,7 +39,7 @@
"path": "clip",
"model_card": "https://github.com/openai/CLIP/blob/main/model-card.md",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_large_patch14/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14/1",
},
"clip_vit_large_patch14_336": {
"metadata": {
Expand All @@ -52,7 +52,7 @@
"path": "clip",
"model_card": "https://github.com/openai/CLIP/blob/main/model-card.md",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_large_patch14_336/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14_336/1",
},
"clip_vit_b_32_laion2b_s34b_b79k": {
"metadata": {
Expand All @@ -65,7 +65,7 @@
"path": "clip",
"model_card": "https://github.com/mlfoundations/open_clip",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/1",
},
"clip_vit_h_14_laion2b_s32b_b79k": {
"metadata": {
Expand All @@ -78,7 +78,7 @@
"path": "clip",
"model_card": "https://github.com/mlfoundations/open_clip",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/1",
},
"clip_vit_g_14_laion2b_s12b_b42k": {
"metadata": {
Expand All @@ -91,7 +91,7 @@
"path": "clip",
"model_card": "https://github.com/mlfoundations/open_clip",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/1",
},
"clip_vit_bigg_14_laion2b_39b_b160k": {
"metadata": {
Expand All @@ -104,6 +104,6 @@
"path": "clip",
"model_card": "https://github.com/mlfoundations/open_clip",
},
"kaggle_handle": "kaggle://kerashub/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/1",
},
}
24 changes: 12 additions & 12 deletions keras_hub/src/models/mit/mit_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b0_ade20k_512/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b0_ade20k_512/2",
},
"mit_b1_ade20k_512": {
"metadata": {
Expand All @@ -32,7 +32,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b1_ade20k_512/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b1_ade20k_512/2",
},
"mit_b2_ade20k_512": {
"metadata": {
Expand All @@ -43,7 +43,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b2_ade20k_512/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b2_ade20k_512/2",
},
"mit_b3_ade20k_512": {
"metadata": {
Expand All @@ -54,7 +54,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b3_ade20k_512/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b3_ade20k_512/2",
},
"mit_b4_ade20k_512": {
"metadata": {
Expand All @@ -65,7 +65,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b4_ade20k_512/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b4_ade20k_512/2",
},
"mit_b5_ade20k_640": {
"metadata": {
Expand All @@ -76,7 +76,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b5_ade20k_640/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b5_ade20k_640/2",
},
"mit_b0_cityscapes_1024": {
"metadata": {
Expand All @@ -87,7 +87,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b0_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b0_cityscapes_1024/2",
},
"mit_b1_cityscapes_1024": {
"metadata": {
Expand All @@ -98,7 +98,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b1_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b1_cityscapes_1024/2",
},
"mit_b2_cityscapes_1024": {
"metadata": {
Expand All @@ -109,7 +109,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b2_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b2_cityscapes_1024/2",
},
"mit_b3_cityscapes_1024": {
"metadata": {
Expand All @@ -120,7 +120,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b3_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b3_cityscapes_1024/2",
},
"mit_b4_cityscapes_1024": {
"metadata": {
Expand All @@ -131,7 +131,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b4_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b4_cityscapes_1024/2",
},
"mit_b5_cityscapes_1024": {
"metadata": {
Expand All @@ -142,7 +142,7 @@
"official_name": "MiT",
"path": "mit",
},
"kaggle_handle": "kaggle://kerashub/mix-transformer/keras/mit_b5_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/mit/keras/mit_b5_cityscapes_1024/2",
},
}

Expand Down
24 changes: 12 additions & 12 deletions keras_hub/src/models/segformer/segformer_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"official_name": "SegFormerB0",
"path": "segformer_b0",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b0_ade20k_512/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b0_ade20k_512/2",
},
"segformer_b1_ade20k_512": {
"metadata": {
Expand All @@ -21,7 +21,7 @@
"official_name": "SegFormerB1",
"path": "segformer_b1",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b1_ade20k_512/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b1_ade20k_512/2",
},
"segformer_b2_ade20k_512": {
"metadata": {
Expand All @@ -32,7 +32,7 @@
"official_name": "SegFormerB2",
"path": "segformer_b2",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b2_ade20k_512/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b2_ade20k_512/2",
},
"segformer_b3_ade20k_512": {
"metadata": {
Expand All @@ -43,7 +43,7 @@
"official_name": "SegFormerB3",
"path": "segformer_b3",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b3_ade20k_512/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b3_ade20k_512/2",
},
"segformer_b4_ade20k_512": {
"metadata": {
Expand All @@ -54,7 +54,7 @@
"official_name": "SegFormerB4",
"path": "segformer_b4",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b4_ade20k_512/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b4_ade20k_512/2",
},
"segformer_b5_ade20k_640": {
"metadata": {
Expand All @@ -65,7 +65,7 @@
"official_name": "SegFormerB5",
"path": "segformer_b5",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b5_ade20k_640/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b5_ade20k_640/2",
},
"segformer_b0_cityscapes_1024": {
"metadata": {
Expand All @@ -76,7 +76,7 @@
"official_name": "SegFormerB0",
"path": "segformer_b0",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b0_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b0_cityscapes_1024/2",
},
"segformer_b1_cityscapes_1024": {
"metadata": {
Expand All @@ -87,7 +87,7 @@
"official_name": "SegFormerB1",
"path": "segformer_b1",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b1_ade20k_512/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b1_ade20k_512/2",
},
"segformer_b2_cityscapes_1024": {
"metadata": {
Expand All @@ -98,7 +98,7 @@
"official_name": "SegFormerB2",
"path": "segformer_b2",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b2_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b2_cityscapes_1024/2",
},
"segformer_b3_cityscapes_1024": {
"metadata": {
Expand All @@ -109,7 +109,7 @@
"official_name": "SegFormerB3",
"path": "segformer_b3",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b3_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b3_cityscapes_1024/2",
},
"segformer_b4_cityscapes_1024": {
"metadata": {
Expand All @@ -120,7 +120,7 @@
"official_name": "SegFormerB4",
"path": "segformer_b4",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b4_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b4_cityscapes_1024/2",
},
"segformer_b5_cityscapes_1024": {
"metadata": {
Expand All @@ -131,6 +131,6 @@
"official_name": "SegFormerB5",
"path": "segformer_b5",
},
"kaggle_handle": "kaggle://kerashub/segformer/keras/segformer_b5_cityscapes_1024/2",
"kaggle_handle": "kaggle://keras/segformer/keras/segformer_b5_cityscapes_1024/2",
},
}
69 changes: 44 additions & 25 deletions tools/hf_uploaded_presets.json
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,14 @@
"kaggle://keras/bert/keras/bert_small_en_uncased/2",
"kaggle://keras/bert/keras/bert_tiny_en_uncased/2",
"kaggle://keras/bert/keras/bert_tiny_en_uncased_sst2/4",
"kaggle://keras/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/1",
"kaggle://keras/clip/keras/clip_vit_base_patch16/1",
"kaggle://keras/clip/keras/clip_vit_base_patch32/1",
"kaggle://keras/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/1",
"kaggle://keras/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/1",
"kaggle://keras/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/1",
"kaggle://keras/clip/keras/clip_vit_large_patch14/1",
"kaggle://keras/clip/keras/clip_vit_large_patch14_336/1",
"kaggle://keras/bloom/keras/bloom_1.1b_multi/1",
"kaggle://keras/bloom/keras/bloom_1.7b_multi/1",
"kaggle://keras/bloom/keras/bloom_3b_multi/1",
Expand All @@ -44,8 +52,18 @@
"kaggle://keras/distil_bert/keras/distil_bert_base_en/2",
"kaggle://keras/distil_bert/keras/distil_bert_base_en_uncased/2",
"kaggle://keras/distil_bert/keras/distil_bert_base_multi/2",
"kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet",
"kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet",
"kaggle://keras/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b2_ra_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b3_ra2_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b4_ra2_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b5_sw_ft_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_b5_sw_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_el_ra_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_em_ra2_imagenet/1",
"kaggle://keras/efficientnet/keras/efficientnet_es_ra_imagenet/1",
"kaggle://keras/electra/keras/electra_base_discriminator_uncased_en/1",
"kaggle://keras/electra/keras/electra_base_generator_uncased_en/1",
"kaggle://keras/electra/keras/electra_large_discriminator_uncased_en/1",
Expand Down Expand Up @@ -83,18 +101,18 @@
"kaggle://keras/mistral/keras/mistral_0.2_instruct_7b_en/1",
"kaggle://keras/mistral/keras/mistral_7b_en/6",
"kaggle://keras/mistral/keras/mistral_instruct_7b_en/6",
"kaggle://keras/mit/keras/mit_b0_ade20k_512/1",
"kaggle://keras/mit/keras/mit_b0_cityscapes_1024/1",
"kaggle://keras/mit/keras/mit_b1_ade20k_512/1",
"kaggle://keras/mit/keras/mit_b1_cityscapes_1024/1",
"kaggle://keras/mit/keras/mit_b2_ade20k_512/1",
"kaggle://keras/mit/keras/mit_b2_cityscapes_1024/1",
"kaggle://keras/mit/keras/mit_b3_ade20k_512/1",
"kaggle://keras/mit/keras/mit_b3_cityscapes_1024/1",
"kaggle://keras/mit/keras/mit_b4_ade20k_512/1",
"kaggle://keras/mit/keras/mit_b4_cityscapes_1024/1",
"kaggle://keras/mit/keras/mit_b5_ade20k_640/1",
"kaggle://keras/mit/keras/mit_b5_cityscapes_1024/1",
"kaggle://keras/mit/keras/mit_b0_ade20k_512/2",
"kaggle://keras/mit/keras/mit_b0_cityscapes_1024/2",
"kaggle://keras/mit/keras/mit_b1_ade20k_512/2",
"kaggle://keras/mit/keras/mit_b1_cityscapes_1024/2",
"kaggle://keras/mit/keras/mit_b2_ade20k_512/2",
"kaggle://keras/mit/keras/mit_b2_cityscapes_1024/2",
"kaggle://keras/mit/keras/mit_b3_ade20k_512/2",
"kaggle://keras/mit/keras/mit_b3_cityscapes_1024/2",
"kaggle://keras/mit/keras/mit_b4_ade20k_512/2",
"kaggle://keras/mit/keras/mit_b4_cityscapes_1024/2",
"kaggle://keras/mit/keras/mit_b5_ade20k_640/2",
"kaggle://keras/mit/keras/mit_b5_cityscapes_1024/2",
"kaggle://keras/opt/keras/opt_1.3b_en/2",
"kaggle://keras/opt/keras/opt_125m_en/2",
"kaggle://keras/opt/keras/opt_2.7b_en/2",
Expand Down Expand Up @@ -122,22 +140,23 @@
"kaggle://keras/resnetv1/keras/resnet_50_imagenet/2",
"kaggle://keras/resnetv2/keras/resnet_v2_101_imagenet/2",
"kaggle://keras/resnetv2/keras/resnet_v2_50_imagenet/2",
"kaggle://keras/retinanet/keras/retinanet_resnet50_fpn_coco/1",
"kaggle://keras/roberta/keras/roberta_base_en/2",
"kaggle://keras/roberta/keras/roberta_large_en/2",
"kaggle://keras/sam/keras/sam_base_sa1b/4",
"kaggle://keras/sam/keras/sam_huge_sa1b/4",
"kaggle://keras/sam/keras/sam_large_sa1b/4",
"kaggle://keras/segformer/keras/segformer_b0_ade20k_512",
"kaggle://keras/segformer/keras/segformer_b0_cityscapes_1024",
"kaggle://keras/segformer/keras/segformer_b1_ade20k_512",
"kaggle://keras/segformer/keras/segformer_b2_ade20k_512",
"kaggle://keras/segformer/keras/segformer_b2_cityscapes_1024",
"kaggle://keras/segformer/keras/segformer_b3_ade20k_512",
"kaggle://keras/segformer/keras/segformer_b3_cityscapes_1024",
"kaggle://keras/segformer/keras/segformer_b4_ade20k_512",
"kaggle://keras/segformer/keras/segformer_b4_cityscapes_1024",
"kaggle://keras/segformer/keras/segformer_b5_ade20k_640",
"kaggle://keras/segformer/keras/segformer_b5_cityscapes_1024",
"kaggle://keras/segformer/keras/segformer_b0_ade20k_512/2",
"kaggle://keras/segformer/keras/segformer_b0_cityscapes_1024/2",
"kaggle://keras/segformer/keras/segformer_b1_ade20k_512/2",
"kaggle://keras/segformer/keras/segformer_b2_ade20k_512/2",
"kaggle://keras/segformer/keras/segformer_b2_cityscapes_1024/2",
"kaggle://keras/segformer/keras/segformer_b3_ade20k_512/2",
"kaggle://keras/segformer/keras/segformer_b3_cityscapes_1024/2",
"kaggle://keras/segformer/keras/segformer_b4_ade20k_512/2",
"kaggle://keras/segformer/keras/segformer_b4_cityscapes_1024/2",
"kaggle://keras/segformer/keras/segformer_b5_ade20k_640/2",
"kaggle://keras/segformer/keras/segformer_b5_cityscapes_1024/2",
"kaggle://keras/stablediffusion-3.5/keras/stable_diffusion_3.5_large/1",
"kaggle://keras/stablediffusion-3.5/keras/stable_diffusion_3.5_large_turbo/1",
"kaggle://keras/stablediffusion3/keras/stable_diffusion_3_medium/3",
Expand Down
Loading
Loading