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

Fix linking error

Browse Source
This commit is contained in:
doctorpangloss 2024-09-25 10:16:56 -07:00
parent 8f58242c91
commit a78f20178d
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy/model_downloader.py
View File

@ -132,20 +132,23 @@ def get_or_download(folder_name: str, filename: str, known_files: Optional[List[
del v del v
logging.info(f"Converted {path} to 16 bit, size is {os.stat(path, follow_symlinks=True).st_size}") logging.info(f"Converted {path} to 16 bit, size is {os.stat(path, follow_symlinks=True).st_size}")
link_exc_info = None link_successful = True
if linked_filename is not None: if linked_filename is not None:
destination_link = os.path.join(this_model_directory, linked_filename) destination_link = os.path.join(this_model_directory, linked_filename)
try: try:
os.makedirs(this_model_directory, exist_ok=True) os.makedirs(this_model_directory, exist_ok=True)
os.symlink(path, destination_link) os.symlink(path, destination_link)
except Exception as exc_info: except Exception as exc_info:
logging.error("error while symbolic linking", exc_info=exc_info)
try: try:
os.link(path, destination_link) os.link(path, destination_link)
except Exception as exc_info: except Exception as hard_link_exc:
logging.error("error while hard linking", exc_info=hard_link_exc)
if cache_hit: if cache_hit:
shutil.copyfile(path, destination_link) shutil.copyfile(path, destination_link)
link_exc_info = exc_info link_successful = False
if link_exc_info is not None:
if not link_successful:
logging.error(f"Failed to link file with alternative download save name in a way that is compatible with Hugging Face caching {repr(known_file)}. If cache_hit={cache_hit} is True, the file was copied into the destination.", exc_info=exc_info) logging.error(f"Failed to link file with alternative download save name in a way that is compatible with Hugging Face caching {repr(known_file)}. If cache_hit={cache_hit} is True, the file was copied into the destination.", exc_info=exc_info)
else: else:
url: Optional[str] = None url: Optional[str] = None