update config - fix gpt/gpt-2 from pretrained
This commit is contained in:
1 parent
6b13f4cb3a
commit
273617b86d
3 files changed
+3
-3
No files matched your search
@@ -423,7 +423,7 @@ class GPT2PreTrainedModel(PreTrainedModel):
|
||||
"""
|
||||
num_special_tokens = kwargs.pop('num_special_tokens', None)
|
||||
|
||||
model = PreTrainedModel.from_pretrained(cls, pretrained_model_name_or_path, *inputs, **kwargs)
|
||||
model = super(PreTrainedModel, cls).from_pretrained(pretrained_model_name_or_path, *inputs, **kwargs)
|
||||
|
||||
# Add additional embeddings for special tokens if needed
|
||||
# This step also make sure we are still sharing the output and input embeddings after loading weights
|
||||
|
||||
Reference in new issue
Block a user