From 9b55ffe449d14b3aed4fe9e272e4e83c962f5d3d Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 5 Jun 2025 14:17:59 +0300 Subject: [PATCH] SDNQ fix VAE x2 --- modules/processing_vae.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/modules/processing_vae.py b/modules/processing_vae.py index 2d3f8bb7a..38e2574cd 100644 --- a/modules/processing_vae.py +++ b/modules/processing_vae.py @@ -140,8 +140,14 @@ def full_vae_decode(latents, model): if shift_factor: latents = latents + shift_factor - if getattr(model.vae, "post_quant_conv", None) is not None and "VAE" not in shared.opts.sdnq_quantize_weights: - latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype) + if getattr(model.vae, "post_quant_conv", None) is not None: + if hasattr(model.vae.post_quant_conv, "bias"): + latents = latents.to(model.vae.post_quant_conv.bias.dtype) + else: + if "VAE" in shared.opts.sdnq_quantize_weights: + latents = latents.to(devices.dtype_vae) + else: + latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype) else: latents = latents.to(model.vae.dtype)