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

Update devices.py - Make 'use-cpu all' actually apply to 'all'
This commit is contained in:
AUTOMATIC1111 2023-12-02 09:46:19 +03:00 committed by GitHub
commit 88736b5557
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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()