From b31d02ba1d6841e8f00462d2dc055a97b1a7d4a1 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 29 Sep 2024 20:28:01 -0400 Subject: [PATCH] cleanup Signed-off-by: Vladimir Mandic --- modules/devices.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/devices.py b/modules/devices.py index 73ee0a94b..e14594d21 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -386,13 +386,13 @@ def set_dtype(): dtype_unet = torch.float16 if opts.no_half: - log.info('Torch override dtype: no-half set') dtype = torch.float32 dtype_vae = torch.float32 dtype_unet = torch.float32 + log.info(f'Torch override: no-half dtype={dtype}') if opts.no_half_vae: - log.info('Torch override VAE dtype: no-half set') dtype_vae = torch.float32 + log.info(f'Torch override: no-half-vae dtype={dtype_vae}') unet_needs_upcast = opts.upcast_sampling if opts.inference_mode == 'inference-mode': inference_context = torch.inference_mode