Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ad8321512d857578ce2e2469eb82a8e8e45c9037"
Commit 44e9ddd7 authored by thomwolf's avatar thomwolf
Browse files

fix num_special_tokens in GPT 2 test

parent cad88e19
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
pytorch_pretrained_bert/modeling_gpt2.py
View file @ 44e9ddd7
......@@ -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]
......
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