Update config.json

config.json

@@ -1 +1,10 @@
-{"architectures": ["BigramLanguageModel"], "n_embd": 64, "n_head": 4, "n_layer": 4, "vocab_size": 61, "block_size": 32, "dropout": 0.1, "model_type": "custom"}
+{
+  "architectures": ["GPT2LMHeadModel"],
+  "n_embd": 64,
+  "n_head": 4,
+  "n_layer": 4,
+  "vocab_size": 61,
+  "block_size": 32,
+  "dropout": 0.1,
+  "model_type": "gpt2"
+}