fix test examples et model pretrained

2019-07-11 22:29:08 +02:00
parent 92a782b108
commit c6bf1a400d
3 changed files with 4 additions and 2 deletions
--- a/pytorch_transformers/modeling_utils.py
+++ b/pytorch_transformers/modeling_utils.py
@@ -49,7 +49,7 @@ class PretrainedConfig(object):
        self.torchscript = kwargs.pop('torchscript', False)

    @classmethod
-    def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
+    def from_pretrained(cls, pretrained_model_name_or_path, *input, **kwargs):
        """
        Instantiate a PretrainedConfig from a pre-trained model configuration.