fix test examples et model pretrained

This commit is contained in:
thomwolf committed 2019-07-11 22:29:08 +02:00
1 parent 92a782b108
commit c6bf1a400d
3 files changed
+4 -2

No files matched your search

+1 -1
View File
@@ -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.