From 490ebbdcf7f6e1171058fded5db6cb231d18636d Mon Sep 17 00:00:00 2001
From: Anish Moorthy <anish.moorthy@worthix.com>
Date: Mon, 22 Jul 2019 15:51:51 -0400
Subject: [PATCH] Fix PretrainedModel.from_pretrained not passing cache_dir
 forward

---
 pytorch_transformers/modeling_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pytorch_transformers/modeling_utils.py b/pytorch_transformers/modeling_utils.py
index a4e1a44c9d..0a4bfa7ba0 100644
--- 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