From 03763cd2db45b763cc3c30c68b6919c3041705b7 Mon Sep 17 00:00:00 2001 From: Michael Bugert Date: Mon, 22 Jul 2019 20:56:51 +0200 Subject: [PATCH] Fixed PreTrainedModel.from_pretrained(...) not passing cache_dir to PretrainedConfig.from_pretrained(...) --- pytorch_transformers/modeling_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_transformers/modeling_utils.py b/pytorch_transformers/modeling_utils.py index 324cdc17c9dd..1a9b8cd43bef 100644 --- a/pytorch_transformers/modeling_utils.py +++ b/pytorch_transformers/modeling_utils.py @@ -359,7 +359,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *inputs, **kwargs): # Load config if config is None: - config = cls.config_class.from_pretrained(pretrained_model_name_or_path, *inputs, **kwargs) + config = cls.config_class.from_pretrained(pretrained_model_name_or_path, cache_dir=cache_dir, *inputs, **kwargs) # Load model if pretrained_model_name_or_path in cls.pretrained_model_archive_map: