robinzixuan commited on
Commit
1bbc668
1 Parent(s): 9f0e988

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -2
config.json CHANGED
@@ -4,7 +4,8 @@
4
  ],
5
  "attention_probs_dropout_prob": 0.0,
6
  "encoder_stride": 16,
7
- "hidden_act": "gelu",
 
8
  "hidden_dropout_prob": 0.0,
9
  "hidden_size": 384,
10
  "id2label": {
@@ -2022,5 +2023,11 @@
2022
  "patch_size": 16,
2023
  "qkv_bias": true,
2024
  "torch_dtype": "float32",
2025
- "transformers_version": "4.41.2"
 
 
 
 
 
 
2026
  }
 
4
  ],
5
  "attention_probs_dropout_prob": 0.0,
6
  "encoder_stride": 16,
7
+ "hidden_act": "gelu"
8
+ ,
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 384,
11
  "id2label": {
 
2023
  "patch_size": 16,
2024
  "qkv_bias": true,
2025
  "torch_dtype": "float32",
2026
+ "transformers_version": "4.41.2",
2027
+ "auto_map":
2028
+ {"AutoConfig": "configuration_vit.ViTConfig",
2029
+ "AutoModel": "modeling_vit.ViTModel",
2030
+ "AutoModelForImageClassification": "modeling_vit.ViTForImageClassification"
2031
+
2032
+ }
2033
  }