Update comfy/model_management.py

comfy/model_management.py

@@ -25,6 +25,8 @@ import sys
 import platform
 import weakref
 import gc
+import os
+os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 
 class VRAMState(Enum):
     DISABLED = 0    #No vram present: no need to move models to vram
@@ -117,16 +119,16 @@ def get_torch_device():
         global directml_device
         return directml_device
     if cpu_state == CPUState.MPS:
-        return torch.device("mps")
+        return torch.device("cpu")
     if cpu_state == CPUState.CPU:
         return torch.device("cpu")
     else:
         if is_intel_xpu():
-            return torch.device("xpu", torch.xpu.current_device())
+            return torch.device("cpu")
         elif is_ascend_npu():
-            return torch.device("npu", torch.npu.current_device())
+            return torch.device("cpu")
         else:
-            return torch.device(torch.cuda.current_device())
+            return torch.device("cpu")
 
 def get_total_memory(dev=None, torch_total_too=False):
     global directml_enabled
@@ -790,7 +792,7 @@ def vae_dtype(device=None, allowed_dtypes=[]):
 def get_autocast_device(dev):
     if hasattr(dev, 'type'):
         return dev.type
-    return "cuda"
+    return "cpu"
 
 def supports_dtype(device, dtype): #TODO
     if dtype == torch.float32: