Update app.py

app.py

@@ -145,11 +145,27 @@ setup_environment()
 
 # --- ComfyUI Imports ---
 # Configure Execution Arguments for ComfyUI
-if not torch.cuda.is_available():
-    print("⚠️ CUDA not available, forcing CPU mode for ComfyUI...")
-    # Inject --cpu argument so ComfyUI optimizes for CPU instead of crashing on CUDA init
-    if "--cpu" not in sys.argv:
-        sys.argv.append("--cpu")
+# Aggressively force CPU if CUDA is not available or if we want to ensure no crashes on CPU Spaces
+try:
+    if not torch.cuda.is_available():
+        print("⚠️ CUDA not available, forcing CPU mode for ComfyUI...")
+        
+        # 1. Force environment variable
+        os.environ["CUDA_VISIBLE_DEVICES"] = ""
+        
+        # 2. Inject --cpu argument
+        if "--cpu" not in sys.argv:
+            sys.argv.append("--cpu")
+
+        # 3. Monkeypatch torch.cuda to ensure ComfyUI doesn't try to initialize CUDA
+        # This is necessary because some ComfyUI versions checks might be aggressive
+        torch.cuda.is_available = lambda: False
+        torch.cuda.device_count = lambda: 0
+        torch.cuda.current_device = lambda: None
+        
+        print("✅ Applied CPU enforcement patches.")
+except Exception as e:
+    print(f"⚠️ Error applying CPU patches: {e}")
 
 # These must happen AFTER setup because ComfyUI folder might not exist before
 try: