Update config.json

2020-03-11 16:12:53 +00:00 · 2020-03-11 16:12:53 +00:00 · e9ce49e76b
parent 9aad4637a4
commit e9ce49e76b
1 changed files with 6 additions and 3 deletions
--- 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,