Spaces:
Runtime error
Runtime error
File size: 3,647 Bytes
bbf161a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | import os
import sys
import shutil
from core.settings import *
def move_and_overwrite(src, dst):
if os.path.isdir(src):
os.makedirs(dst, exist_ok=True)
for item in os.listdir(src):
s_child = os.path.join(src, item)
d_child = os.path.join(dst, item)
move_and_overwrite(s_child, d_child)
elif os.path.isfile(src):
if os.path.exists(dst):
try:
os.remove(dst)
except OSError:
pass
shutil.move(src, dst)
def initialize_comfyui():
APP_DIR = sys.path[0]
COMFYUI_TEMP_DIR = "ComfyUI_temp"
print("--- Cloning ComfyUI Repository ---")
if not os.path.exists(COMFYUI_TEMP_DIR):
os.system(f"git clone https://github.com/comfy-Org/ComfyUI {COMFYUI_TEMP_DIR}")
print("✅ ComfyUI repository cloned.")
else:
print("✅ ComfyUI repository already exists.")
print(f"--- Merging ComfyUI from '{COMFYUI_TEMP_DIR}' to '{APP_DIR}' ---")
for item in os.listdir(COMFYUI_TEMP_DIR):
src_path = os.path.join(COMFYUI_TEMP_DIR, item)
dst_path = os.path.join(APP_DIR, item)
if item == '.git':
continue
move_and_overwrite(src_path, dst_path)
try:
shutil.rmtree(COMFYUI_TEMP_DIR)
print("✅ ComfyUI merged and temporary directory removed.")
except OSError as e:
print(f"⚠️ Could not remove temporary directory '{COMFYUI_TEMP_DIR}': {e}")
print(f"✅ Current working directory is: {os.getcwd()}")
# [ZeroGPU / Upstream ComfyUI Compatibility]
# Guard against ComfyUI PR #16746 is_integrated attribute check crash under ZeroGPU SimpleNamespace
try:
import torch
if hasattr(torch, "cuda") and hasattr(torch.cuda, "get_device_properties"):
_orig_gdp = torch.cuda.get_device_properties
def _safe_gdp(*args, **kwargs):
p = _orig_gdp(*args, **kwargs)
if not hasattr(p, "is_integrated"):
try:
p.is_integrated = False
except Exception:
pass
return p
torch.cuda.get_device_properties = _safe_gdp
except Exception:
pass
mm_path = os.path.join(APP_DIR, "comfy", "model_management.py")
if os.path.exists(mm_path):
try:
with open(mm_path, "r", encoding="utf-8") as f:
mm_code = f.read()
unsafe_check = "bool(torch.cuda.get_device_properties(device).is_integrated)"
safe_check = "bool(getattr(torch.cuda.get_device_properties(device), 'is_integrated', False))"
if unsafe_check in mm_code:
with open(mm_path, "w", encoding="utf-8") as f:
f.write(mm_code.replace(unsafe_check, safe_check))
print("🛡️ [Setup] Patched comfy/model_management.py for safe is_integrated attribute access.")
except Exception as e:
print(f"⚠️ [Setup] Failed to patch comfy/model_management.py: {e}")
import comfy.model_management
print("--- Environment Ready ---")
print("✅ ComfyUI initialized with default attention mechanism.")
for dir_path in CATEGORY_TO_DIR_MAP.values():
os.makedirs(os.path.join(APP_DIR, dir_path), exist_ok=True)
os.makedirs(os.path.join(APP_DIR, INPUT_DIR), exist_ok=True)
os.makedirs(os.path.join(APP_DIR, OUTPUT_DIR), exist_ok=True)
print("✅ All required model directories are present.") |