Skip to content

Commit 96b8685

Browse files
authored
Update metadata (#2026)
* Fixes for preset loading * Update all json, fix a few preset loading bugs
1 parent 7058174 commit 96b8685

34 files changed

+209
-203
lines changed

keras_hub/src/models/albert/albert_presets.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"params": 11683584,
1111
"path": "albert",
1212
},
13-
"kaggle_handle": "kaggle://keras/albert/keras/albert_base_en_uncased/2",
13+
"kaggle_handle": "kaggle://keras/albert/keras/albert_base_en_uncased/5",
1414
},
1515
"albert_large_en_uncased": {
1616
"metadata": {
@@ -21,7 +21,7 @@
2121
"params": 17683968,
2222
"path": "albert",
2323
},
24-
"kaggle_handle": "kaggle://keras/albert/keras/albert_large_en_uncased/2",
24+
"kaggle_handle": "kaggle://keras/albert/keras/albert_large_en_uncased/3",
2525
},
2626
"albert_extra_large_en_uncased": {
2727
"metadata": {
@@ -32,7 +32,7 @@
3232
"params": 58724864,
3333
"path": "albert",
3434
},
35-
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_large_en_uncased/2",
35+
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_large_en_uncased/3",
3636
},
3737
"albert_extra_extra_large_en_uncased": {
3838
"metadata": {
@@ -43,6 +43,6 @@
4343
"params": 222595584,
4444
"path": "albert",
4545
},
46-
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_extra_large_en_uncased/2",
46+
"kaggle_handle": "kaggle://keras/albert/keras/albert_extra_extra_large_en_uncased/3",
4747
},
4848
}

keras_hub/src/models/bart/bart_presets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"params": 139417344,
1111
"path": "bart",
1212
},
13-
"kaggle_handle": "kaggle://keras/bart/keras/bart_base_en/2",
13+
"kaggle_handle": "kaggle://keras/bart/keras/bart_base_en/3",
1414
},
1515
"bart_large_en": {
1616
"metadata": {
@@ -30,7 +30,7 @@
3030
"dropout": 0.1,
3131
"max_sequence_length": 1024,
3232
},
33-
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en/2",
33+
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en/3",
3434
},
3535
"bart_large_en_cnn": {
3636
"metadata": {
@@ -50,6 +50,6 @@
5050
"dropout": 0.1,
5151
"max_sequence_length": 1024,
5252
},
53-
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en_cnn/2",
53+
"kaggle_handle": "kaggle://keras/bart/keras/bart_large_en_cnn/3",
5454
},
5555
}

keras_hub/src/models/bert/bert_presets.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"params": 4385920,
1111
"path": "bert",
1212
},
13-
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased/2",
13+
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased/3",
1414
},
1515
"bert_small_en_uncased": {
1616
"metadata": {
@@ -21,7 +21,7 @@
2121
"params": 28763648,
2222
"path": "bert",
2323
},
24-
"kaggle_handle": "kaggle://keras/bert/keras/bert_small_en_uncased/2",
24+
"kaggle_handle": "kaggle://keras/bert/keras/bert_small_en_uncased/3",
2525
},
2626
"bert_medium_en_uncased": {
2727
"metadata": {
@@ -32,7 +32,7 @@
3232
"params": 41373184,
3333
"path": "bert",
3434
},
35-
"kaggle_handle": "kaggle://keras/bert/keras/bert_medium_en_uncased/2",
35+
"kaggle_handle": "kaggle://keras/bert/keras/bert_medium_en_uncased/3",
3636
},
3737
"bert_base_en_uncased": {
3838
"metadata": {
@@ -43,7 +43,7 @@
4343
"params": 109482240,
4444
"path": "bert",
4545
},
46-
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en_uncased/2",
46+
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en_uncased/3",
4747
},
4848
"bert_base_en": {
4949
"metadata": {
@@ -54,7 +54,7 @@
5454
"params": 108310272,
5555
"path": "bert",
5656
},
57-
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en/2",
57+
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_en/3",
5858
},
5959
"bert_base_zh": {
6060
"metadata": {
@@ -64,7 +64,7 @@
6464
"params": 102267648,
6565
"path": "bert",
6666
},
67-
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_zh/2",
67+
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_zh/3",
6868
},
6969
"bert_base_multi": {
7070
"metadata": {
@@ -75,7 +75,7 @@
7575
"params": 177853440,
7676
"path": "bert",
7777
},
78-
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_multi/2",
78+
"kaggle_handle": "kaggle://keras/bert/keras/bert_base_multi/3",
7979
},
8080
"bert_large_en_uncased": {
8181
"metadata": {
@@ -86,7 +86,7 @@
8686
"params": 335141888,
8787
"path": "bert",
8888
},
89-
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en_uncased/2",
89+
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en_uncased/3",
9090
},
9191
"bert_large_en": {
9292
"metadata": {
@@ -97,7 +97,7 @@
9797
"params": 333579264,
9898
"path": "bert",
9999
},
100-
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en/2",
100+
"kaggle_handle": "kaggle://keras/bert/keras/bert_large_en/3",
101101
},
102102
"bert_tiny_en_uncased_sst2": {
103103
"metadata": {
@@ -108,6 +108,6 @@
108108
"params": 4385920,
109109
"path": "bert",
110110
},
111-
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased_sst2/4",
111+
"kaggle_handle": "kaggle://keras/bert/keras/bert_tiny_en_uncased_sst2/5",
112112
},
113113
}

keras_hub/src/models/bloom/bloom_presets.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"params": 559214592,
1111
"path": "bloom",
1212
},
13-
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_560m_multi/3",
13+
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_560m_multi/4",
1414
},
1515
"bloom_1.1b_multi": {
1616
"metadata": {
@@ -21,7 +21,7 @@
2121
"params": 1065314304,
2222
"path": "bloom",
2323
},
24-
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.1b_multi/1",
24+
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.1b_multi/2",
2525
},
2626
"bloom_1.7b_multi": {
2727
"metadata": {
@@ -32,7 +32,7 @@
3232
"params": 1722408960,
3333
"path": "bloom",
3434
},
35-
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.7b_multi/1",
35+
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_1.7b_multi/2",
3636
},
3737
"bloom_3b_multi": {
3838
"metadata": {
@@ -43,7 +43,7 @@
4343
"params": 3002557440,
4444
"path": "bloom",
4545
},
46-
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_3b_multi/1",
46+
"kaggle_handle": "kaggle://keras/bloom/keras/bloom_3b_multi/2",
4747
},
4848
"bloomz_560m_multi": {
4949
"metadata": {
@@ -54,7 +54,7 @@
5454
"params": 559214592,
5555
"path": "bloom",
5656
},
57-
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_560m_multi/1",
57+
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_560m_multi/2",
5858
},
5959
"bloomz_1.1b_multi": {
6060
"metadata": {
@@ -65,7 +65,7 @@
6565
"params": 1065314304,
6666
"path": "bloom",
6767
},
68-
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.1b_multi/1",
68+
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.1b_multi/2",
6969
},
7070
"bloomz_1.7b_multi": {
7171
"metadata": {
@@ -76,7 +76,7 @@
7676
"params": 1722408960,
7777
"path": "bloom",
7878
},
79-
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.7b_multi/1",
79+
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_1.7b_multi/2",
8080
},
8181
"bloomz_3b_multi": {
8282
"metadata": {
@@ -87,6 +87,6 @@
8787
"params": 3002557440,
8888
"path": "bloom",
8989
},
90-
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_3b_multi/1",
90+
"kaggle_handle": "kaggle://keras/bloom/keras/bloomz_3b_multi/2",
9191
},
9292
}

keras_hub/src/models/clip/clip_presets.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
"params": 149620934,
1212
"path": "clip",
1313
},
14-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch16/1",
14+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch16/2",
1515
},
1616
"clip_vit_base_patch32": {
1717
"metadata": {
@@ -22,7 +22,7 @@
2222
"params": 151277363,
2323
"path": "clip",
2424
},
25-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch32/1",
25+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_base_patch32/2",
2626
},
2727
"clip_vit_large_patch14": {
2828
"metadata": {
@@ -33,7 +33,7 @@
3333
"params": 427616770,
3434
"path": "clip",
3535
},
36-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14/1",
36+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14/2",
3737
},
3838
"clip_vit_large_patch14_336": {
3939
"metadata": {
@@ -44,7 +44,7 @@
4444
"params": 427944770,
4545
"path": "clip",
4646
},
47-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14_336/1",
47+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_large_patch14_336/2",
4848
},
4949
"clip_vit_b_32_laion2b_s34b_b79k": {
5050
"metadata": {
@@ -55,7 +55,7 @@
5555
"params": 151277363,
5656
"path": "clip",
5757
},
58-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/1",
58+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_b_32_laion2b_s34b_b79k/2",
5959
},
6060
"clip_vit_h_14_laion2b_s32b_b79k": {
6161
"metadata": {
@@ -66,7 +66,7 @@
6666
"params": 986109698,
6767
"path": "clip",
6868
},
69-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/1",
69+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_h_14_laion2b_s32b_b79k/2",
7070
},
7171
"clip_vit_g_14_laion2b_s12b_b42k": {
7272
"metadata": {
@@ -77,7 +77,7 @@
7777
"params": 1366678530,
7878
"path": "clip",
7979
},
80-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/1",
80+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_g_14_laion2b_s12b_b42k/2",
8181
},
8282
"clip_vit_bigg_14_laion2b_39b_b160k": {
8383
"metadata": {
@@ -88,6 +88,6 @@
8888
"params": 2539567362,
8989
"path": "clip",
9090
},
91-
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/1",
91+
"kaggle_handle": "kaggle://keras/clip/keras/clip_vit_bigg_14_laion2b_39b_b160k/2",
9292
},
9393
}

keras_hub/src/models/deberta_v3/deberta_v3_presets.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"params": 70682112,
1111
"path": "deberta_v3",
1212
},
13-
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_extra_small_en/2",
13+
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_extra_small_en/3",
1414
},
1515
"deberta_v3_small_en": {
1616
"metadata": {
@@ -21,7 +21,7 @@
2121
"params": 141304320,
2222
"path": "deberta_v3",
2323
},
24-
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_small_en/2",
24+
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_small_en/3",
2525
},
2626
"deberta_v3_base_en": {
2727
"metadata": {
@@ -32,7 +32,7 @@
3232
"params": 183831552,
3333
"path": "deberta_v3",
3434
},
35-
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_en/2",
35+
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_en/3",
3636
},
3737
"deberta_v3_large_en": {
3838
"metadata": {
@@ -43,7 +43,7 @@
4343
"params": 434012160,
4444
"path": "deberta_v3",
4545
},
46-
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_large_en/2",
46+
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_large_en/3",
4747
},
4848
"deberta_v3_base_multi": {
4949
"metadata": {
@@ -54,6 +54,6 @@
5454
"params": 278218752,
5555
"path": "deberta_v3",
5656
},
57-
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_multi/2",
57+
"kaggle_handle": "kaggle://keras/deberta_v3/keras/deberta_v3_base_multi/3",
5858
},
5959
}

keras_hub/src/models/deeplab_v3/deeplab_v3_presets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,6 @@
1212
"params": 39190656,
1313
"path": "deeplab_v3",
1414
},
15-
"kaggle_handle": "kaggle://keras/deeplabv3plus/keras/deeplab_v3_plus_resnet50_pascalvoc/3",
15+
"kaggle_handle": "kaggle://keras/deeplabv3plus/keras/deeplab_v3_plus_resnet50_pascalvoc/4",
1616
},
1717
}

keras_hub/src/models/densenet/densenet_presets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"params": 7037504,
1111
"path": "densenet",
1212
},
13-
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/2",
13+
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_121_imagenet/3",
1414
},
1515
"densenet_169_imagenet": {
1616
"metadata": {
@@ -21,7 +21,7 @@
2121
"params": 12642880,
2222
"path": "densenet",
2323
},
24-
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/2",
24+
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_169_imagenet/3",
2525
},
2626
"densenet_201_imagenet": {
2727
"metadata": {
@@ -32,6 +32,6 @@
3232
"params": 18321984,
3333
"path": "densenet",
3434
},
35-
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/2",
35+
"kaggle_handle": "kaggle://keras/densenet/keras/densenet_201_imagenet/3",
3636
},
3737
}

keras_hub/src/models/distil_bert/distil_bert_presets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
"params": 66362880,
1212
"path": "distil_bert",
1313
},
14-
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en_uncased/2",
14+
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en_uncased/3",
1515
},
1616
"distil_bert_base_en": {
1717
"metadata": {
@@ -23,7 +23,7 @@
2323
"params": 65190912,
2424
"path": "distil_bert",
2525
},
26-
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en/2",
26+
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_en/3",
2727
},
2828
"distil_bert_base_multi": {
2929
"metadata": {
@@ -34,6 +34,6 @@
3434
"params": 134734080,
3535
"path": "distil_bert",
3636
},
37-
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_multi/2",
37+
"kaggle_handle": "kaggle://keras/distil_bert/keras/distil_bert_base_multi/3",
3838
},
3939
}

0 commit comments

Comments
 (0)