From 50949fab74d7d40ad4057f694ce843dae5349d60 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 10 Aug 2022 15:46:03 -0400 Subject: [PATCH] Properly move cache when it is not in default path (#18563) --- src/transformers/utils/hub.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/transformers/utils/hub.py b/src/transformers/utils/hub.py index 00f9c277c41773..163ad64ffa173b 100644 --- a/src/transformers/utils/hub.py +++ b/src/transformers/utils/hub.py @@ -1070,7 +1070,11 @@ def move_cache(cache_dir=None, new_cache_dir=None, token=None): "`transformers.utils.move_cache()`." ) try: - move_cache() + if TRANSFORMERS_CACHE != default_cache_path: + # Users set some env variable to customize cache storage + move_cache(TRANSFORMERS_CACHE, TRANSFORMERS_CACHE) + else: + move_cache() except Exception as e: trace = "\n".join(traceback.format_tb(e.__traceback__)) logger.error(