Skip to content

Commit 47e1676

Browse files
authored
New cache fixes: add safeguard before looking in folders (#18522)
1 parent 7495924 commit 47e1676

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/transformers/utils/hub.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,8 @@ def get_cached_models(cache_dir: Union[str, Path] = None) -> List[Tuple]:
133133
cache_dir = TRANSFORMERS_CACHE
134134
elif isinstance(cache_dir, Path):
135135
cache_dir = str(cache_dir)
136+
if not os.path.isdir(cache_dir):
137+
return []
136138

137139
cached_models = []
138140
for file in os.listdir(cache_dir):
@@ -210,6 +212,9 @@ def try_to_load_from_cache(cache_dir, repo_id, filename, revision=None):
210212
if not os.path.isdir(model_cache):
211213
# No cache for this model
212214
return None
215+
for subfolder in ["refs", "snapshots"]:
216+
if not os.path.isdir(os.path.join(model_cache, subfolder)):
217+
return None
213218

214219
# Resolve refs (for instance to convert main to the associated commit sha)
215220
cached_refs = os.listdir(os.path.join(model_cache, "refs"))
@@ -873,6 +878,8 @@ def get_all_cached_files(cache_dir=None):
873878
cache_dir = TRANSFORMERS_CACHE
874879
else:
875880
cache_dir = str(cache_dir)
881+
if not os.path.isdir(cache_dir):
882+
return []
876883

877884
cached_files = []
878885
for file in os.listdir(cache_dir):

0 commit comments

Comments
 (0)