diff --git a/fix_torch.py b/fix_torch.py new file mode 100644 index 00000000..b60a0bbb --- /dev/null +++ b/fix_torch.py @@ -0,0 +1,16 @@ +import importlib.util +import shutil +import os +import ctypes +import logging + + +torch_spec = importlib.util.find_spec("torch") +for folder in torch_spec.submodule_search_locations: + lib_folder = os.path.join(folder, "lib") + test_file = os.path.join(lib_folder, "fbgemm.dll") + try: + mydll = ctypes.cdll.LoadLibrary(test_file) + except FileNotFoundError as e: + logging.warning("Detected pytorch version with libomp issue, patching.") + shutil.copyfile(os.path.join(lib_folder, "libiomp5md.dll"), os.path.join(lib_folder, "libomp140.x86_64.dll")) diff --git a/main.py b/main.py index 196351a3..479643b8 100644 --- a/main.py +++ b/main.py @@ -74,6 +74,12 @@ if __name__ == "__main__": import cuda_malloc +if args.windows_standalone_build: + try: + import fix_torch + except: + pass + import comfy.utils import yaml