Skip to content

Commit ea524b9

Browse files
xywei00rwightman
authored andcommitted
add missing ViTamin configs
1 parent 8328e89 commit ea524b9

File tree

2 files changed

+40
-0
lines changed

2 files changed

+40
-0
lines changed
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
{
2+
"embed_dim": 768,
3+
"vision_cfg": {
4+
"timm_model_name": "vitamin_large_384",
5+
"timm_model_pretrained": false,
6+
"timm_pool": "",
7+
"timm_proj": "linear",
8+
"timm_drop": 0.0,
9+
"timm_drop_path": 0.1,
10+
"image_size": 384
11+
},
12+
"text_cfg": {
13+
"context_length": 77,
14+
"vocab_size": 49408,
15+
"width": 768,
16+
"heads": 12,
17+
"layers": 12
18+
},
19+
"custom_text": true
20+
}
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
{
2+
"embed_dim": 1024,
3+
"vision_cfg": {
4+
"timm_model_name": "vitamin_large2_384",
5+
"timm_model_pretrained": false,
6+
"timm_pool": "",
7+
"timm_proj": "linear",
8+
"timm_drop": 0.0,
9+
"timm_drop_path": 0.1,
10+
"image_size": 384
11+
},
12+
"text_cfg": {
13+
"context_length": 77,
14+
"vocab_size": 49408,
15+
"width": 1024,
16+
"heads": 16,
17+
"layers": 24
18+
},
19+
"custom_text": true
20+
}

0 commit comments

Comments
 (0)