From ab756f713c7dd5257e27bb74edd906dfdfbf0e5d Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 13 Mar 2020 21:00:54 +0100 Subject: [PATCH] add gpt2-xl for tf --- src/transformers/modeling_tf_gpt2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/modeling_tf_gpt2.py b/src/transformers/modeling_tf_gpt2.py index 9111f7a923..4225270664 100644 --- a/src/transformers/modeling_tf_gpt2.py +++ b/src/transformers/modeling_tf_gpt2.py @@ -40,6 +40,7 @@ TF_GPT2_PRETRAINED_MODEL_ARCHIVE_MAP = { "gpt2": "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-tf_model.h5", "gpt2-medium": "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-medium-tf_model.h5", "gpt2-large": "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-large-tf_model.h5", + "gpt2-xl": "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-xl-tf_model.h5", "distilgpt2": "https://s3.amazonaws.com/models.huggingface.co/bert/distilgpt2-tf_model.h5", }