From cf94a4425b68aac956e5d5bf126180fe0c49382e Mon Sep 17 00:00:00 2001
From: Lysandre <lysandre.debut@reseau.eseo.fr>
Date: Thu, 17 Feb 2022 22:28:17 +0000
Subject: [PATCH] Update config.json

---
 config.json | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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"
+  ]
 }