Skip to content

GitLab

Unverified Commit b3f9e945 authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #687 from huggingface/tests_and_doc 

Updating tests and doc
parents cad88e19 44e9ddd7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
pytorch_pretrained_bert/modeling_gpt2.py
View file @ b3f9e945
...@@ -432,6 +432,8 @@ class GPT2PreTrainedModel(nn.Module): ...@@ -432,6 +432,8 @@ class GPT2PreTrainedModel(nn.Module):
kwargs.pop('cache_dir', None) kwargs.pop('cache_dir', None)
from_tf = kwargs.get('from_tf', False) from_tf = kwargs.get('from_tf', False)
kwargs.pop('from_tf', None) 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: if pretrained_model_name_or_path in PRETRAINED_MODEL_ARCHIVE_MAP:
archive_file = PRETRAINED_MODEL_ARCHIVE_MAP[pretrained_model_name_or_path] archive_file = PRETRAINED_MODEL_ARCHIVE_MAP[pretrained_model_name_or_path]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment