Merge pull request #14131 from read-0nly/patch-1

Update devices.py - Make 'use-cpu all' actually apply to 'all'

Merge pull request #14131 from read-0nly/patch-1
Update devices.py - Make 'use-cpu all' actually apply to 'all'
88736b55 · AUTOMATIC1111 · GitHub · 9eadc4f1 · 3cd6e1d0 · 88736b55
Commit 88736b55 authored Dec 02, 2023 by AUTOMATIC1111 Committed by GitHub Dec 02, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/devices.py modules/devices.py +1 -1

No files found.
--- 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()