Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Change vae cast order

Browse Source
This commit is contained in:
Disty0
2024-08-21 23:46:54 +03:00
parent 963940b9ae
commit 3c4d9f313d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+3 -3
View File
@@ -45,10 +45,10 @@ def full_vae_decode(latents, model):
if upcast: # this is done by diffusers automatically if output_type != 'latent'
model.upcast_vae()
if shared.opts.no_half_vae:
latents = latents.to(torch.float32)
elif getattr(model.vae, "post_quant_conv", None) is not None:
if getattr(model.vae, "post_quant_conv", None) is not None:
latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype)
elif shared.opts.no_half_vae:
latents = latents.to(torch.float32)
# normalize latents
latents_mean = model.vae.config.get("latents_mean", None)