Merge branch 'KoboldAI:main' into united

2022-07-27 18:06:53 +02:00 · 2022-07-27 18:06:53 +02:00 · 699c2353e7
parent fe64e480ee e8a077f072
commit 699c2353e7
1 changed files with 1 additions and 1 deletions
--- a/utils.py
+++ b/utils.py
@ -210,7 +210,7 @@ def aria2_hook(pretrained_model_name_or_path: str, force_download=False, cache_d
    def is_cached(url):
        try:
            transformers.file_utils.get_from_cache(url, cache_dir=cache_dir, local_files_only=True)
-        except FileNotFoundError:
+        except (FileNotFoundError, transformers.file_utils.EntryNotFoundError):
            return False
        return True
    while True:  # Try to get the huggingface.co URL of the model's pytorch_model.bin or pytorch_model.bin.index.json file