From 50e62a4cb4d503e3559b88838b8cf9f745fef516 Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Thu, 11 Jul 2019 16:50:21 -0400 Subject: [PATCH] fix gpt/gpt-2 from pretrained --- pytorch_transformers/modeling_gpt2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_transformers/modeling_gpt2.py b/pytorch_transformers/modeling_gpt2.py index 29d1cbae42..495e002529 100644 --- a/pytorch_transformers/modeling_gpt2.py +++ b/pytorch_transformers/modeling_gpt2.py @@ -423,7 +423,7 @@ class GPT2PreTrainedModel(PreTrainedModel): """ num_special_tokens = kwargs.pop('num_special_tokens', None) - model = super(PreTrainedModel, cls).from_pretrained(pretrained_model_name_or_path, *inputs, **kwargs) + model = super().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