Add max length

fdc487d8 · thomwolf · aa05dc89 · fdc487d8
Commit fdc487d8 authored Aug 21, 2019 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

pytorch_transformers/tokenization_gpt2.py pytorch_transformers/tokenization_gpt2.py +1 -0

No files found.
--- a/pytorch_transformers/tokenization_gpt2.py
+++ b/pytorch_transformers/tokenization_gpt2.py
@@ -58,6 +58,7 @@ PRETRAINED_VOCAB_FILES_MAP = {
 PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES = {
    'gpt2': 1024,
    'gpt2-medium': 1024,
+    'gpt2-large': 1024,
 }

 @lru_cache()