Skip to content

Commit 50949fa

Browse files
authored
Properly move cache when it is not in default path (#18563)
1 parent 6936e7c commit 50949fa

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/transformers/utils/hub.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1070,7 +1070,11 @@ def move_cache(cache_dir=None, new_cache_dir=None, token=None):
10701070
"`transformers.utils.move_cache()`."
10711071
)
10721072
try:
1073-
move_cache()
1073+
if TRANSFORMERS_CACHE != default_cache_path:
1074+
# Users set some env variable to customize cache storage
1075+
move_cache(TRANSFORMERS_CACHE, TRANSFORMERS_CACHE)
1076+
else:
1077+
move_cache()
10741078
except Exception as e:
10751079
trace = "\n".join(traceback.format_tb(e.__traceback__))
10761080
logger.error(

0 commit comments

Comments
 (0)