Commit 88736b55 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub

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

Update devices.py - Make 'use-cpu all' actually apply to 'all'
parents 9eadc4f1 3cd6e1d0
...@@ -38,7 +38,7 @@ def get_optimal_device(): ...@@ -38,7 +38,7 @@ def get_optimal_device():
def get_device_for(task): 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 cpu
return get_optimal_device() return get_optimal_device()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment