add gpt2-xl for tf

ab756f71 · Patrick von Platen · Julien Chaumond · cc4c3795 · ab756f71
Commit ab756f71 authored Mar 13, 2020 by Patrick von Platen Committed by Julien Chaumond Mar 13, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/modeling_tf_gpt2.py src/transformers/modeling_tf_gpt2.py +1 -0

No files found.
--- 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",
 }