robinzixuan commited on
Commit
a59be4f
1 Parent(s): 96e0175

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -2,6 +2,12 @@
2
  "architectures": [
3
  "ViTForImageClassification"
4
  ],
 
 
 
 
 
 
5
  "attention_probs_dropout_prob": 0.0,
6
  "encoder_stride": 16,
7
  "hidden_act": "gelu",
 
2
  "architectures": [
3
  "ViTForImageClassification"
4
  ],
5
+ "auto_map":
6
+ {"AutoConfig": "configuration_vit.ViTConfig",
7
+ "AutoModel": "modeling_vit.ViTModel",
8
+ "AutoModelForImageClassification": "modeling_vit.ViTForImageClassification"
9
+
10
+ },
11
  "attention_probs_dropout_prob": 0.0,
12
  "encoder_stride": 16,
13
  "hidden_act": "gelu",