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

Fix torch 2.4.0 fix

Browse Source
This commit is contained in:
doctorpangloss 2024-07-30 11:17:25 -07:00
parent 249b60e354
commit 4e851e2cfa
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
comfy/cmd/main_pre.py
View File

@ -59,19 +59,22 @@ def _fix_pytorch_240():
if os.path.exists(dest):
break
with open(test_file, 'rb') as f:
contents = f.read()
# todo: dubious
if b"libomp140.x86_64.dll" not in contents:
break
try:
_ = ctypes.cdll.LoadLibrary(test_file)
except FileNotFoundError:
logging.warning("Detected pytorch version with libomp issue, trying to patch")
with open(test_file, 'rb') as f:
contents = f.read()
# todo: dubious
if b"libomp140.x86_64.dll" not in contents:
break
try:
shutil.copyfile(os.path.join(lib_folder, "libiomp5md.dll"), dest)
except Exception as exc_info:
logging.error("While trying to patch a fix for torch 2.4.0, an error occurred, which means this is unlikely to work", exc_info=exc_info)
_ = ctypes.cdll.LoadLibrary(test_file)
except FileNotFoundError:
logging.warning("Detected pytorch version with libomp issue, trying to patch")
try:
shutil.copyfile(os.path.join(lib_folder, "libiomp5md.dll"), dest)
except Exception as exc_info:
logging.error("While trying to patch a fix for torch 2.4.0, an error occurred, which means this is unlikely to work", exc_info=exc_info)
except:
pass
def _create_tracer():