From 9532f8931f6f9464c6ddd4d3b182dcc836da4f16 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 12 Feb 2024 21:53:51 +0300 Subject: [PATCH] Fix vae has no post_quant_conv --- modules/processing_vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/processing_vae.py b/modules/processing_vae.py index bb0a11cb5..55c545cad 100644 --- a/modules/processing_vae.py +++ b/modules/processing_vae.py @@ -44,8 +44,8 @@ def full_vae_decode(latents, model): upcast = (model.vae.dtype == torch.float16) and getattr(model.vae.config, 'force_upcast', False) and hasattr(model, 'upcast_vae') if upcast: # this is done by diffusers automatically if output_type != 'latent' model.upcast_vae() - - latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype) + if hasattr(model.vae, "post_quant_conv"): + latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype) decoded = model.vae.decode(latents / model.vae.config.scaling_factor, return_dict=False)[0] # Delete PyTorch VAE after OpenVINO compile