Properly move cache when it is not in default path (#18563)

50949fab · Sylvain Gugger · GitHub · 6936e7c4 · 50949fab
Unverified Commit 50949fab authored Aug 10, 2022 by Sylvain Gugger Committed by GitHub Aug 10, 2022
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

src/transformers/utils/hub.py src/transformers/utils/hub.py +5 -1

No files found.
--- a/src/transformers/utils/hub.py
+++ b/src/transformers/utils/hub.py
@@ -1070,6 +1070,10 @@ if cache_version < 1:
            "`transformers.utils.move_cache()`."
        )
    try:
+        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__))