diff --git a/config.json b/config.json
index 5de2e4d..a9b135b 100644
--- a/config.json
+++ b/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"
+  ]
 }