Update config.json
Browse files- 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 |
}
|