Fix PretrainedModel.from_pretrained not passing cache_dir forward

2019-07-22 15:51:51 -04:00
parent b8009cb0da
commit 490ebbdcf7
1 changed files with 2 additions and 1 deletions
--- a/pytorch_transformers/modeling_utils.py
+++ b/pytorch_transformers/modeling_utils.py
@@ -373,7 +373,8 @@ class PreTrainedModel(nn.Module):
        if config is None:
            config, model_kwargs = cls.config_class.from_pretrained(
                pretrained_model_name_or_path, *model_args,
-                return_unused_args=True, **kwargs
+                cache_dir=cache_dir, return_unused_args=True,
+                **kwargs
            )
        else:
            model_kwargs = kwargs