fix test examples et model pretrained
This commit is contained in:
@@ -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.
|
||||
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user