diff --git a/config.json b/config.json
index f4878c6..e50ba4d 100644
--- a/config.json
+++ b/config.json
@@ -1,8 +1,11 @@
 {
+	"architectures": [
+		"GPT2LMHeadModel"
+	],
 	"bos_token_id": 50256,
 	"eos_token_ids": [
-	50256
-	  ],
+		50256
+	],
 	"initializer_range": 0.02,
 	"layer_norm_epsilon": 1e-05,
 	"model_type": "gpt2",