update config - fix gpt/gpt-2 from pretrained

2019-07-11 22:45:03 +02:00
parent 6b13f4cb3a
commit 273617b86d
3 changed files with 3 additions and 3 deletions
--- 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 = 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