Skip to content

Commit

Permalink
New cache fixes: add safeguard before looking in folders (huggingface…
Browse files Browse the repository at this point in the history
  • Loading branch information
sgugger authored and amyeroberts committed Aug 17, 2022
1 parent dcb1685 commit 7072f66
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions src/transformers/utils/hub.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,6 +133,8 @@ def get_cached_models(cache_dir: Union[str, Path] = None) -> List[Tuple]:
cache_dir = TRANSFORMERS_CACHE
elif isinstance(cache_dir, Path):
cache_dir = str(cache_dir)
if not os.path.isdir(cache_dir):
return []

cached_models = []
for file in os.listdir(cache_dir):
Expand Down Expand Up @@ -210,6 +212,9 @@ def try_to_load_from_cache(cache_dir, repo_id, filename, revision=None):
if not os.path.isdir(model_cache):
# No cache for this model
return None
for subfolder in ["refs", "snapshots"]:
if not os.path.isdir(os.path.join(model_cache, subfolder)):
return None

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

cached_files = []
for file in os.listdir(cache_dir):
Expand Down

0 comments on commit 7072f66

Please sign in to comment.