From f75f77adb4413ee961c25e1f8df84a08cba08ea4 Mon Sep 17 00:00:00 2001 From: system Date: Wed, 11 Mar 2020 16:12:55 +0000 Subject: [PATCH] Update config.json --- config.json | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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",