Commit 490ebbdc authored by Anish Moorthy's avatar Anish Moorthy
Browse files

Fix PretrainedModel.from_pretrained not passing cache_dir forward

parent b8009cb0
...@@ -373,7 +373,8 @@ class PreTrainedModel(nn.Module): ...@@ -373,7 +373,8 @@ class PreTrainedModel(nn.Module):
if config is None: if config is None:
config, model_kwargs = cls.config_class.from_pretrained( config, model_kwargs = cls.config_class.from_pretrained(
pretrained_model_name_or_path, *model_args, pretrained_model_name_or_path, *model_args,
return_unused_args=True, **kwargs cache_dir=cache_dir, return_unused_args=True,
**kwargs
) )
else: else:
model_kwargs = kwargs model_kwargs = kwargs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment