Transformers
PyTorch
English
bridgetower
Inference Endpoints
anahita-b commited on
Commit
f812c88
1 Parent(s): e7129ac

Update vit model name and add image_size

Browse files
Files changed (1) hide show
  1. 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-16-weights.pt",
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,