Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

SDNQ fix VAE x2

Browse Source
This commit is contained in:
Disty0
2025-06-05 14:17:59 +03:00
parent ad2a4ad616
commit 9b55ffe449
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+8 -2
View File
@@ -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)