diff options
| author | FNSpd <125805478+FNSpd@users.noreply.github.com> | 2023-03-21 14:45:39 +0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-21 14:45:39 +0400 |
| commit | 254d9946439be9b6266b671db68086eb68ef1e63 (patch) | |
| tree | be424037987513479951e623b0ecdb56bff0a53c /modules/devices.py | |
| parent | a9fed7c364061ae6efb37f797b6b522cb3cf7aa2 (diff) | |
Update devices.py
Diffstat (limited to 'modules/devices.py')
| -rw-r--r-- | modules/devices.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/devices.py b/modules/devices.py index 52c3e7cd..6c6c7233 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -110,7 +110,7 @@ def autocast(disable=False): if disable: return contextlib.nullcontext() - if dtype == torch.float32 or shared.cmd_opts.precision == "full": + if dtype == torch.float32 or shared.cmd_opts.precision == "full" or shared.cmd_opts.upcast_sampling: return contextlib.nullcontext() return torch.autocast("cuda") |