Merge pull request #687 from huggingface/tests_and_doc

Updating tests and doc

Merge pull request #687 from huggingface/tests_and_doc
Updating tests and doc
b3f9e945 · Thomas Wolf · GitHub · cad88e19 · 44e9ddd7 · b3f9e945
Unverified Commit b3f9e945 authored Jun 14, 2019 by Thomas Wolf Committed by GitHub Jun 14, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

pytorch_pretrained_bert/modeling_gpt2.py pytorch_pretrained_bert/modeling_gpt2.py +2 -0

No files found.
--- a/pytorch_pretrained_bert/modeling_gpt2.py
+++ b/pytorch_pretrained_bert/modeling_gpt2.py
@@ -432,6 +432,8 @@ class GPT2PreTrainedModel(nn.Module):
        kwargs.pop('cache_dir', None)
        from_tf = kwargs.get('from_tf', False)
        kwargs.pop('from_tf', None)
+        num_special_tokens = kwargs.get('num_special_tokens', None)
+        kwargs.pop('num_special_tokens', None)

        if pretrained_model_name_or_path in PRETRAINED_MODEL_ARCHIVE_MAP:
            archive_file = PRETRAINED_MODEL_ARCHIVE_MAP[pretrained_model_name_or_path]