marcusinthesky commited on
Commit
3e521f9
1 Parent(s): 8003a5d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -9,7 +9,7 @@
9
  "AutoModel": "modelling.VTDE"
10
  },
11
  "logit_scale_init_value": 2.6592,
12
- "model_type": "vision-text-dual-encoder",
13
  "projection_dim": 384,
14
  "text_config": {
15
  "_name_or_path": "intfloat/e5-small-v2",
@@ -177,5 +177,5 @@
177
  "use_bfloat16": false,
178
  "use_swiglu_ffn": false
179
  },
180
- "vision_pooling_mode": "max"
181
  }
 
9
  "AutoModel": "modelling.VTDE"
10
  },
11
  "logit_scale_init_value": 2.6592,
12
+ "model_type": "vtde",
13
  "projection_dim": 384,
14
  "text_config": {
15
  "_name_or_path": "intfloat/e5-small-v2",
 
177
  "use_bfloat16": false,
178
  "use_swiglu_ffn": false
179
  },
180
+ "vision_pooling_mode": "norm"
181
  }