aboutsummaryrefslogtreecommitdiff
path: root/modules/devices.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-10-15 10:35:18 +0300
committerGitHub <noreply@github.com>2022-10-15 10:35:18 +0300
commitaf144ebdc70ef6f006bebd47af7a8d3180665538 (patch)
treec3ef9ef64f9959f2820a8083420f4f9b7486d636 /modules/devices.py
parente21f01f64504bc651da6e85216474bbd35ee010d (diff)
parent6a4e84671016d38c10a55fedcdf09321dba737ae (diff)
Merge branch 'master' into ckpt-cache
Diffstat (limited to 'modules/devices.py')
-rw-r--r--modules/devices.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/devices.py b/modules/devices.py
index 03ef58f1..eb422583 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -34,7 +34,7 @@ def enable_tf32():
errors.run(enable_tf32, "Enabling TF32")
-device = device_gfpgan = device_bsrgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device()
+device = device_interrogate = device_gfpgan = device_bsrgan = device_esrgan = device_scunet = device_codeformer = get_optimal_device()
dtype = torch.float16
dtype_vae = torch.float16