From 938a8585f200e725c075aa277d1df8be8eb44d9f Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 25 Sep 2024 18:02:47 -0400 Subject: [PATCH] fix vae upcast condition --- modules/processing_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/processing_vae.py b/modules/processing_vae.py index 68057700d..60e8617dd 100644 --- a/modules/processing_vae.py +++ b/modules/processing_vae.py @@ -52,7 +52,7 @@ def full_vae_decode(latents, model): elif shared.opts.diffusers_offload_mode != "sequential": sd_models.move_model(model.vae, devices.device) - upcast = ((model.vae.dtype == torch.float16) and getattr(model.vae.config, 'force_upcast', False)) or shared.opts.no_half_vae + upcast = (model.vae.dtype == torch.float16) and (getattr(model.vae.config, 'force_upcast', False) or shared.opts.no_half_vae) if upcast: if hasattr(model, 'upcast_vae'): # this is done by diffusers automatically if output_type != 'latent' model.upcast_vae()