marcusinthesky commited on
Commit
a886642
1 Parent(s): 49b1e28

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -4,8 +4,12 @@
4
  "architectures": [
5
  "VTDE"
6
  ],
 
 
 
 
7
  "logit_scale_init_value": 2.6592,
8
- "model_type": "vision-text-dual-encoder",
9
  "projection_dim": 384,
10
  "text_config": {
11
  "_name_or_path": "intfloat/e5-small-v2",
 
4
  "architectures": [
5
  "VTDE"
6
  ],
7
+ "auto_map": {
8
+ "AutoConfig": "configuration_vtde.VTDEConfig",
9
+ "AutoModelForImageClassification": "modeling_vtde.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",