Update config.json

2022-02-17 22:28:17 +00:00 · 2022-02-17 22:28:17 +00:00 · cf94a4425b
parent f7dbfaa8b1
commit cf94a4425b
1 changed files with 4 additions and 1 deletions
--- 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"
+  ]
 }