Update config.json

2020-01-31 23:00:25 +00:00 · 2020-01-31 23:00:25 +00:00 · 6934902484
parent 934da44f9c
commit 6934902484
1 changed files with 3 additions and 0 deletions
--- a/config.json
+++ b/config.json
@ -1,4 +1,7 @@
 {
+  "architectures": [
+    "GPT2LMHeadModel"
+  ],
  "attn_pdrop": 0.1,
  "embd_pdrop": 0.1,
  "finetuning_task": null,