Stop throwing cache warning

2025-10-20 17:13:56 +08:00 · 2024-06-28 15:04:55 +02:00
1 changed files with 3 additions and 3 deletions
--- a/src/transformers/utils/hub.py
+++ b/src/transformers/utils/hub.py
@ -1292,8 +1292,10 @@ class PushInProgress:


 cache_version_file = os.path.join(TRANSFORMERS_CACHE, "version.txt")
+cache_is_not_empty = os.path.isdir(TRANSFORMERS_CACHE) and len(os.listdir(TRANSFORMERS_CACHE)) > 0
+
 if not os.path.isfile(cache_version_file):
-    cache_version = 0
+    cache_version = 0 if cache_is_not_empty else 1
 else:
    with open(cache_version_file) as f:
        try:
@ -1301,8 +1303,6 @@ else:
        except ValueError:
            cache_version = 0

-cache_is_not_empty = os.path.isdir(TRANSFORMERS_CACHE) and len(os.listdir(TRANSFORMERS_CACHE)) > 0
-
 if cache_version < 1 and cache_is_not_empty:
    if is_offline_mode():
        logger.warning(