Update devices.py

fixes issue where "--use-cpu" all properly makes SD run on CPU but leaves ControlNet (and other extensions, I presume) pointed at GPU, causing a crash in ControlNet caused by a mismatch between devices between SD and CN AUTOMATIC1111#14097
ruchej · Sep 30, 2024 · 6180a63 · 6180a63
1 parent 7e4376b
commit 6180a63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/devices.py b/modules/devices.py
@@ -38,7 +38,7 @@ def get_optimal_device():
 
 
 def get_device_for(task):
-    if task in shared.cmd_opts.use_cpu:
+    if task in shared.cmd_opts.use_cpu or "all" in shared.cmd_opts.use_cpu:
         return cpu
 
     return get_optimal_device()