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

cleanups

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-01 20:02:21 -04:00
parent f9b5a83b49
commit 9a46d381cc
8 changed files with 47 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+2 -2
View File
@@ -60,8 +60,7 @@ def full_vae_decode(latents, model):
model.vae.orig_dtype = model.vae.dtype
model.vae = model.vae.to(dtype=torch.float32)
latents = latents.to(torch.float32)
else:
latents = latents.to(devices.device)
latents = latents.to(devices.device)
if getattr(model.vae, "post_quant_conv", None) is not None:
latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype)
@@ -84,6 +83,7 @@ def full_vae_decode(latents, model):
latents_stats = f'shape={latents.shape} dtype={latents.dtype} device={latents.device}'
stats = f'vae {vae_stats} latents {latents_stats}'
log_debug(f'VAE config: {model.vae.config}')
try:
decoded = model.vae.decode(latents, return_dict=False)[0]
except Exception as e: