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

Better interaction with huggingface caches

Browse Source
This commit is contained in:
doctorpangloss 2025-08-07 13:19:20 -07:00
parent 232a5bd5dc
commit b42b2322fa
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_downloader.py
View File

@ -94,7 +94,6 @@ def get_or_download(folder_name: str, filename: str, known_files: Optional[List[
return path return path
with comfy_tqdm(): with comfy_tqdm():
if isinstance(known_file, HuggingFile): if isinstance(known_file, HuggingFile):
symlinks_supported = are_symlinks_supported()
if known_file.save_with_filename is not None: if known_file.save_with_filename is not None:
linked_filename = known_file.save_with_filename linked_filename = known_file.save_with_filename
elif not known_file.force_save_in_repo_id and os.path.basename(known_file.filename) != known_file.filename: elif not known_file.force_save_in_repo_id and os.path.basename(known_file.filename) != known_file.filename:
@ -122,6 +121,8 @@ def get_or_download(folder_name: str, filename: str, known_files: Optional[List[
file_size = None file_size = None
if os.path.isfile(path) and file_size == known_file.size: if os.path.isfile(path) and file_size == known_file.size:
return path return path
# at this point, the file was not found with its candidate name
path = None
cache_hit = False cache_hit = False
try: try:
@ -131,6 +132,7 @@ def get_or_download(folder_name: str, filename: str, known_files: Optional[List[
repo_type=known_file.repo_type, repo_type=known_file.repo_type,
revision=known_file.revision, revision=known_file.revision,
local_files_only=True, local_files_only=True,
local_dir=hf_destination_dir if args.force_hf_local_dir_mode else None,
) )
logger.debug(f"hf_hub_download cache hit for {known_file.repo_id}/{known_file.filename}") logger.debug(f"hf_hub_download cache hit for {known_file.repo_id}/{known_file.filename}")
cache_hit = True cache_hit = True
@ -141,6 +143,7 @@ def get_or_download(folder_name: str, filename: str, known_files: Optional[List[
filename=known_file.filename, filename=known_file.filename,
repo_type=known_file.repo_type, repo_type=known_file.repo_type,
revision=known_file.revision, revision=known_file.revision,
local_dir=hf_destination_dir if args.force_hf_local_dir_mode else None,
) )
except IOError as exc_info: except IOError as exc_info:
logger.error(f"cannot reach huggingface {known_file.repo_id}/{known_file.filename}", exc_info=exc_info) logger.error(f"cannot reach huggingface {known_file.repo_id}/{known_file.filename}", exc_info=exc_info)