diff --git a/config.json b/config.json
index eade54b..9f7102d 100644
--- a/config.json
+++ b/config.json
@@ -57,6 +57,7 @@
     "pad_token_id": 1,
     "prefix": null,
     "problem_type": null,
+    "projection_dim" : 512,
     "pruned_heads": {},
     "remove_invalid_values": false,
     "repetition_penalty": 1.0,
@@ -131,6 +132,7 @@
     "patch_size": 16,
     "prefix": null,
     "problem_type": null,
+    "projection_dim" : 512,
     "pruned_heads": {},
     "remove_invalid_values": false,
     "repetition_penalty": 1.0,