Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"architectures": ["
|
| 3 |
"hidden_size": 768,
|
| 4 |
"num_hidden_layers": 12,
|
| 5 |
"num_attention_heads": 12,
|
|
|
|
| 1 |
{
|
| 2 |
+
"architectures": ["ViTForImageClassification"],
|
| 3 |
"hidden_size": 768,
|
| 4 |
"num_hidden_layers": 12,
|
| 5 |
"num_attention_heads": 12,
|