Update vit model name and add image_size
Browse files- preprocessor_config.json +2 -1
preprocessor_config.json
CHANGED
|
@@ -11,6 +11,7 @@
|
|
| 11 |
"freeze_layer_count_vit":false,
|
| 12 |
"head_hidden_scale":2,
|
| 13 |
"hidden_size":768,
|
|
|
|
| 14 |
"input_text_embed_size":768,
|
| 15 |
"link_tower_shared":false,
|
| 16 |
"link_tower_type":"add",
|
|
@@ -44,7 +45,7 @@
|
|
| 44 |
"unfreeze_RoBERTa_layernorm":false,
|
| 45 |
"unfreeze_ViT_attention":false,
|
| 46 |
"unfreeze_ViT_layernorm":false,
|
| 47 |
-
"vit":"ViT-B
|
| 48 |
"vit_layernorm_init_from_vit":false,
|
| 49 |
"vit_layernorm_shared":true,
|
| 50 |
"vit_remove_last":false,
|
|
|
|
| 11 |
"freeze_layer_count_vit":false,
|
| 12 |
"head_hidden_scale":2,
|
| 13 |
"hidden_size":768,
|
| 14 |
+
"image_size":288,
|
| 15 |
"input_text_embed_size":768,
|
| 16 |
"link_tower_shared":false,
|
| 17 |
"link_tower_type":"add",
|
|
|
|
| 45 |
"unfreeze_RoBERTa_layernorm":false,
|
| 46 |
"unfreeze_ViT_attention":false,
|
| 47 |
"unfreeze_ViT_layernorm":false,
|
| 48 |
+
"vit":"ViT-B/16",
|
| 49 |
"vit_layernorm_init_from_vit":false,
|
| 50 |
"vit_layernorm_shared":true,
|
| 51 |
"vit_remove_last":false,
|