DarthReca commited on
Commit
acef00a
·
verified ·
1 Parent(s): e8e89b9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -15,5 +15,9 @@
15
  "torch_dtype": "float32",
16
  "transformers_version": "4.47.1",
17
  "use_location_encoder": false,
18
- "vision_model_key": "vit_small_patch16_224"
 
 
 
 
19
  }
 
15
  "torch_dtype": "float32",
16
  "transformers_version": "4.47.1",
17
  "use_location_encoder": false,
18
+ "vision_model_key": "vit_small_patch16_224",
19
+ "auto_map": {
20
+ "AutoModel": "modeling_closp.CLOSPModel",
21
+ "AutoConfig": "modeling_closp.CLOSPConfig"
22
+ }
23
  }