Unverified Commit a9eab236 authored by FNSpd's avatar FNSpd Committed by GitHub
Browse files

Update devices.py

parent 280ed8f0

modules/devices.py

+1 −1
Original line number Diff line number Diff line
@@ -110,7 +110,7 @@ def autocast(disable=False): 
    if disable:

        return contextlib.nullcontext()



    if dtype == torch.float32 or shared.cmd_opts.precision == "full" or shared.cmd_opts.upcast_sampling:

    if dtype == torch.float32 or shared.cmd_opts.precision == "full":

        return contextlib.nullcontext()



    return torch.autocast("cuda")

CRA Git | Maintained and supported by SUSTech CRA and CCSE