From 3a97db1f8a746e829625ccd57bd4b2003edb20ef Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 22 Aug 2024 00:16:00 +0300 Subject: [PATCH] Fix setting no-half-vae --- modules/sd_models.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 0b99ec9af..88a27a58e 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -678,10 +678,11 @@ def set_diffuser_options(sd_model, vae = None, op: str = 'model', offload=True): shared.log.debug(f'Setting {op} VAE: name={sd_vae.loaded_vae_file}') if shared.opts.diffusers_vae_upcast != 'default': sd_model.vae.config.force_upcast = True if shared.opts.diffusers_vae_upcast == 'true' else False - if shared.opts.no_half_vae: - devices.dtype_vae = torch.float32 - sd_model.vae.to(devices.dtype_vae) shared.log.debug(f'Setting {op} VAE: upcast={sd_model.vae.config.force_upcast}') + if shared.opts.no_half_vae: + devices.dtype_vae = torch.float32 + sd_model.vae.to(devices.dtype_vae) + shared.log.debug(f'Setting {op} VAE: no-half') if hasattr(sd_model, "enable_vae_slicing"): if shared.opts.diffusers_vae_slicing: shared.log.debug(f'Setting {op}: enable VAE slicing')