Skip to content

Commit

Permalink
Update all json, fix a few preset loading bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
mattdangerw committed Dec 21, 2024
1 parent 4d792a3 commit 5160d0b
Show file tree
Hide file tree
Showing 31 changed files with 188 additions and 188 deletions.
8 changes: 4 additions & 4 deletions keras_hub/src/models/albert/albert_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"params": 11683584,
"path": "albert",
},
"kaggle_handle": "kaggle://keras/albert/keras/albert_base_en_uncased/2",
"kaggle_handle": "kaggle://keras/albert/keras/albert_base_en_uncased/5",
},
"albert_large_en_uncased": {
"metadata": {
Expand All @@ -21,7 +21,7 @@
"params": 17683968,
"path": "albert",
},
"kaggle_handle": "kaggle://keras/albert/keras/albert_large_en_uncased/2",
"kaggle_handle": "kaggle://keras/albert/keras/albert_large_en_uncased/3",
},
"albert_extra_large_en_uncased": {
"metadata": {
Expand All @@ -32,7 +32,7 @@
"params": 58724864,
"path": "albert",
},
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_large_en_uncased/2",
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_large_en_uncased/3",
},
"albert_extra_extra_large_en_uncased": {
"metadata": {
Expand All @@ -43,6 +43,6 @@
"params": 222595584,
"path": "albert",
},
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_extra_large_en_uncased/2",
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_extra_large_en_uncased/3",
},
}
6 changes: 3 additions & 3 deletions keras_hub/src/models/bart/bart_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"params": 139417344,
"path": "bart",
},
"kaggle_handle": "kaggle://keras/bart/keras/bart_base_en/2",
"kaggle_handle": "kaggle://keras/bart/keras/bart_base_en/3",
},
"bart_large_en": {
"metadata": {
Expand All @@ -30,7 +30,7 @@
"dropout": 0.1,
"max_sequence_length": 1024,
},
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en/2",
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en/3",
},
"bart_large_en_cnn": {
"metadata": {
Expand All @@ -50,6 +50,6 @@
"dropout": 0.1,
"max_sequence_length": 1024,
},
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en_cnn/2",
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en_cnn/3",
},
}
20 changes: 10 additions & 10 deletions keras_hub/src/models/bert/bert_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"params": 4385920,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased/3",
},
"bert_small_en_uncased": {
"metadata": {
Expand All @@ -21,7 +21,7 @@
"params": 28763648,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_small_en_uncased/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_small_en_uncased/3",
},
"bert_medium_en_uncased": {
"metadata": {
Expand All @@ -32,7 +32,7 @@
"params": 41373184,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_medium_en_uncased/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_medium_en_uncased/3",
},
"bert_base_en_uncased": {
"metadata": {
Expand All @@ -43,7 +43,7 @@
"params": 109482240,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en_uncased/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en_uncased/3",
},
"bert_base_en": {
"metadata": {
Expand All @@ -54,7 +54,7 @@
"params": 108310272,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en/3",
},
"bert_base_zh": {
"metadata": {
Expand All @@ -64,7 +64,7 @@
"params": 102267648,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_zh/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_zh/3",
},
"bert_base_multi": {
"metadata": {
Expand All @@ -75,7 +75,7 @@
"params": 177853440,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_multi/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_multi/3",
},
"bert_large_en_uncased": {
"metadata": {
Expand All @@ -86,7 +86,7 @@
"params": 335141888,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en_uncased/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en_uncased/3",
},
"bert_large_en": {
"metadata": {
Expand All @@ -97,7 +97,7 @@
"params": 333579264,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en/2",
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en/3",
},
"bert_tiny_en_uncased_sst2": {
"metadata": {
Expand All @@ -108,6 +108,6 @@
"params": 4385920,
"path": "bert",
},
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased_sst2/4",
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased_sst2/5",
},
}
16 changes: 8 additions & 8 deletions keras_hub/src/models/bloom/bloom_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"params": 559214592,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_560m_multi/3",
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_560m_multi/4",
},
"bloom_1.1b_multi": {
"metadata": {
Expand All @@ -21,7 +21,7 @@
"params": 1065314304,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.1b_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.1b_multi/2",
},
"bloom_1.7b_multi": {
"metadata": {
Expand All @@ -32,7 +32,7 @@
"params": 1722408960,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.7b_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.7b_multi/2",
},
"bloom_3b_multi": {
"metadata": {
Expand All @@ -43,7 +43,7 @@
"params": 3002557440,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_3b_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_3b_multi/2",
},
"bloomz_560m_multi": {
"metadata": {
Expand All @@ -54,7 +54,7 @@
"params": 559214592,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_560m_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_560m_multi/2",
},
"bloomz_1.1b_multi": {
"metadata": {
Expand All @@ -65,7 +65,7 @@
"params": 1065314304,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.1b_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.1b_multi/2",
},
"bloomz_1.7b_multi": {
"metadata": {
Expand All @@ -76,7 +76,7 @@
"params": 1722408960,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.7b_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.7b_multi/2",
},
"bloomz_3b_multi": {
"metadata": {
Expand All @@ -87,6 +87,6 @@
"params": 3002557440,
"path": "bloom",
},
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_3b_multi/1",
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_3b_multi/2",
},
}
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 @@ -11,7 +11,7 @@
"params": 149620934,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch16/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch16/2",
},
"clip_vit_base_patch32": {
"metadata": {
Expand All @@ -22,7 +22,7 @@
"params": 151277363,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch32/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch32/2",
},
"clip_vit_large_patch14": {
"metadata": {
Expand All @@ -33,7 +33,7 @@
"params": 427616770,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14/2",
},
"clip_vit_large_patch14_336": {
"metadata": {
Expand All @@ -44,7 +44,7 @@
"params": 427944770,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14_336/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14_336/2",
},
"clip_vit_b_32_laion2b_s34b_b79k": {
"metadata": {
Expand All @@ -55,7 +55,7 @@
"params": 151277363,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/2",
},
"clip_vit_h_14_laion2b_s32b_b79k": {
"metadata": {
Expand All @@ -66,7 +66,7 @@
"params": 986109698,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/2",
},
"clip_vit_g_14_laion2b_s12b_b42k": {
"metadata": {
Expand All @@ -77,7 +77,7 @@
"params": 1366678530,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/2",
},
"clip_vit_bigg_14_laion2b_39b_b160k": {
"metadata": {
Expand All @@ -88,6 +88,6 @@
"params": 2539567362,
"path": "clip",
},
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/1",
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/2",
},
}
10 changes: 5 additions & 5 deletions keras_hub/src/models/deberta_v3/deberta_v3_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"params": 70682112,
"path": "deberta_v3",
},
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_extra_small_en/2",
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_extra_small_en/3",
},
"deberta_v3_small_en": {
"metadata": {
Expand All @@ -21,7 +21,7 @@
"params": 141304320,
"path": "deberta_v3",
},
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_small_en/2",
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_small_en/3",
},
"deberta_v3_base_en": {
"metadata": {
Expand All @@ -32,7 +32,7 @@
"params": 183831552,
"path": "deberta_v3",
},
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_en/2",
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_en/3",
},
"deberta_v3_large_en": {
"metadata": {
Expand All @@ -43,7 +43,7 @@
"params": 434012160,
"path": "deberta_v3",
},
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_large_en/2",
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_large_en/3",
},
"deberta_v3_base_multi": {
"metadata": {
Expand All @@ -54,6 +54,6 @@
"params": 278218752,
"path": "deberta_v3",
},
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_multi/2",
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_multi/3",
},
}
2 changes: 1 addition & 1 deletion keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,6 @@
"params": 39190656,
"path": "deeplab_v3",
},
"kaggle_handle": "kaggle://keras/deeplabv3plus/keras/deeplab_v3_plus_resnet50_pascalvoc/3",
"kaggle_handle": "kaggle://keras/deeplabv3plus/keras/deeplab_v3_plus_resnet50_pascalvoc/4",
},
}
6 changes: 3 additions & 3 deletions keras_hub/src/models/densenet/densenet_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"params": 7037504,
"path": "densenet",
},
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/2",
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/3",
},
"densenet_169_imagenet": {
"metadata": {
Expand All @@ -21,7 +21,7 @@
"params": 12642880,
"path": "densenet",
},
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/2",
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/3",
},
"densenet_201_imagenet": {
"metadata": {
Expand All @@ -32,6 +32,6 @@
"params": 18321984,
"path": "densenet",
},
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/2",
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/3",
},
}
6 changes: 3 additions & 3 deletions keras_hub/src/models/distil_bert/distil_bert_presets.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
"params": 66362880,
"path": "distil_bert",
},
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en_uncased/2",
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en_uncased/3",
},
"distil_bert_base_en": {
"metadata": {
Expand All @@ -23,7 +23,7 @@
"params": 65190912,
"path": "distil_bert",
},
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en/2",
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en/3",
},
"distil_bert_base_multi": {
"metadata": {
Expand All @@ -34,6 +34,6 @@
"params": 134734080,
"path": "distil_bert",
},
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_multi/2",
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_multi/3",
},
}
Loading

0 comments on commit 5160d0b

Please sign in to comment.