mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-01-09 02:31:57 -08:00
Merge branch 'dev' into test-fp8
This commit is contained in:
commit
110485d5bb
26 changed files with 677 additions and 253 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue