Unverified Commit 88736b55 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

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

Update devices.py - Make 'use-cpu all' actually apply to 'all'
parents 9eadc4f1 3cd6e1d0

modules/devices.py

+1 −1
Original line number Diff line number Diff line
@@ -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()

CRA Git | Maintained and supported by SUSTech CRA and CCSE