diff --git a/config.json b/config.json
index 322d475..830e880 100644
--- a/config.json
+++ b/config.json
@@ -59,6 +59,7 @@
     "pad_token_id": 1,
     "prefix": null,
     "problem_type": null,
+    "projection_dim" : 768,
     "pruned_heads": {},
     "remove_invalid_values": false,
     "repetition_penalty": 1.0,
@@ -139,6 +140,7 @@
     "patch_size": 14,
     "prefix": null,
     "problem_type": null,
+    "projection_dim" : 768,
     "pruned_heads": {},
     "remove_invalid_values": false,
     "repetition_penalty": 1.0,