From e9ce49e76b22ffdcd40b5308331795bb348b26cc Mon Sep 17 00:00:00 2001
From: system <system@huggingface.co>
Date: Wed, 11 Mar 2020 16:12:53 +0000
Subject: [PATCH] Update config.json

---
 config.json | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/config.json b/config.json
index 2914d88..640f914 100644
--- a/config.json
+++ b/config.json
@@ -1,13 +1,16 @@
 {
+	"architectures": [
+		"GPT2LMHeadModel"
+	],
 	"bos_token_id": 50256,
 	"eos_token_ids": [
-	50256
-	  ],
+		50256
+	],
 	"initializer_range": 0.02,
 	"layer_norm_epsilon": 1e-05,
 	"model_type": "gpt2",
 	"n_ctx": 1024,
-	"n_embd": 768,
+	"n_embd": 1024,
 	"n_head": 16,
 	"n_layer": 24,
 	"n_positions": 1024,