aboutsummaryrefslogtreecommitdiffstats
path: root/modules/devices.py
diff options
context:
space:
mode:
authorKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-12-02 09:00:09 +0000
committerKohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>2023-12-02 09:00:09 +0000
commit110485d5bb511ab01ac3d890f1deca0502f4c7db (patch)
tree4335df77c422985e28b30c67ea846ca71e5ef1d7 /modules/devices.py
parent3d341ebc7dcb44df3b4c013b3805c08d8a35e24a (diff)
parent0bb6e00ba3c6775b17b3449c454c6efabb1bdad1 (diff)
downloadstable-diffusion-webui-gfx803-110485d5bb511ab01ac3d890f1deca0502f4c7db.tar.gz
stable-diffusion-webui-gfx803-110485d5bb511ab01ac3d890f1deca0502f4c7db.tar.bz2
stable-diffusion-webui-gfx803-110485d5bb511ab01ac3d890f1deca0502f4c7db.zip
Merge branch 'dev' into test-fp8
Diffstat (limited to 'modules/devices.py')
-rw-r--r--modules/devices.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/devices.py b/modules/devices.py
index c19a7f40..04669b46 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -55,7 +55,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()