Clean setup.py: single CPU fallback patch and torch monkey‑patch before import

comfy_integration/setup.py

@@ -39,7 +39,6 @@ def initialize_comfyui():
     except OSError as e:
         print(f"⚠️ Could not remove temporary directory '{COMFYUI_TEMP_DIR}': {e}")
 
-
     print("--- Cloning third-party extensions for ComfyUI ---")
 
     # 1. ComfyUI_IPAdapter_plus
@@ -84,14 +83,17 @@ def initialize_comfyui():
 
     print(f"✅ Current working directory is: {os.getcwd()}")
     
-    import comfy.model_management as model_mgmt
-    # Patch get_torch_device to ensure CPU fallback on systems without a GPU.
+    # Ensure torch treats CUDA as unavailable on CPU‑only environment.
     import torch
+    torch.cuda.is_available = lambda: False
+
+    import comfy.model_management as model_mgmt
+    # Patch get_torch_device to enforce CPU fallback.
     def _cpu_fallback_get_torch_device():
         """Return appropriate torch device, falling back to CPU when needed.
 +
 +        This patch replaces the original implementation that assumed a CUDA GPU
-+        and raised a RuntimeError on CPU‑only systems.  It respects the existing
++        and raised a RuntimeError on CPU‑only systems. It respects the existing
 +        ``directml_enabled`` and ``cpu_state`` flags, and falls back to CPU when
 +        no GPU backend is available.
 +        """
@@ -111,19 +113,17 @@ def initialize_comfyui():
             return torch.device("npu", torch.npu.current_device())
         if model_mgmt.is_mlu():
             return torch.device("mlu", torch.mlu.current_device())
-        # CUDA – use if available.
+        # CUDA – use if available (will be False due to monkey‑patch).
         if torch.cuda.is_available():
             return torch.device(torch.cuda.current_device())
-        # No GPU detected; default to CPU.
+        # Default to CPU.
         return torch.device("cpu")
     model_mgmt.get_torch_device = _cpu_fallback_get_torch_device
-    
-    # Ensure the patched function is used by subsequent imports.
     import importlib
     importlib.reload(model_mgmt)
     
     print("--- Environment Ready ---")
-
+    
     print("✅ ComfyUI initialized with default attention mechanism.")
 
     for dir_path in CATEGORY_TO_DIR_MAP.values():
@@ -132,4 +132,4 @@ def initialize_comfyui():
     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.")
+    print("✅ All required model directories are present.")