fix test examples et model pretrained

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

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.

View File

@@ -30,6 +30,7 @@ class ModelUtilsTest(unittest.TestCase):
self.assertIsNotNone(config)
self.assertIsInstance(config, PretrainedConfig)
model = BertModel.from_pretrained(model_name)
model, loading_info = BertModel.from_pretrained(model_name, output_loading_info=True)
self.assertIsNotNone(model)
self.assertIsInstance(model, PreTrainedModel)