wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-27 23:00:20 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Tweak downloader test

Browse Source
This commit is contained in:
doctorpangloss 2024-07-09 22:02:07 -07:00
parent 3d67224937
commit 34c9482cfb
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/downloader/test_huggingface_downloads.py
View File

@ -141,7 +141,8 @@ def test_known_repos(tmp_path_factory):
finally: finally:
_delete_repo_from_huggingface_cache(test_repo_id) _delete_repo_from_huggingface_cache(test_repo_id)
_delete_repo_from_huggingface_cache(test_repo_id, test_cache_dir) _delete_repo_from_huggingface_cache(test_repo_id, test_cache_dir)
KNOWN_HUGGINGFACE_MODEL_REPOS.remove(test_repo_id) if test_repo_id in KNOWN_HUGGINGFACE_MODEL_REPOS:
KNOWN_HUGGINGFACE_MODEL_REPOS.remove(test_repo_id)
folder_paths.folder_names_and_paths["huggingface"] = prev_huggingface folder_paths.folder_names_and_paths["huggingface"] = prev_huggingface
folder_paths.folder_names_and_paths["huggingface_cache"] = prev_huggingface_cache folder_paths.folder_names_and_paths["huggingface_cache"] = prev_huggingface_cache
if prev_hub_cache is None and "HF_HUB_CACHE" in os.environ: if prev_hub_cache is None and "HF_HUB_CACHE" in os.environ: