Update config.json

config.json

@@ -13,5 +13,8 @@
   "num_hidden_layers": 5,
   "patch_size": 2,
   "transformers_version": "4.11.0.dev0",
-  "vocab_size": {}
+  "vocab_size": {},
+  "architectures": [
+    "ViTForImageClassification"
+  ]
 }