DarthReca commited on
Commit
9e6a3b8
·
verified ·
1 Parent(s): 552da56

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": true,
18
- "vision_model_key": "resnet50"
 
 
 
 
19
  }
 
15
  "torch_dtype": "float32",
16
  "transformers_version": "4.47.1",
17
  "use_location_encoder": true,
18
+ "vision_model_key": "resnet50",
19
+ "auto_map": {
20
+ "AutoModel": "modeling_closp.CLOSPModel",
21
+ "AutoConfig": "modeling_closp.CLOSPConfig"
22
+ }
23
  }