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

fix vae dtype during load

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-13 08:57:24 -04:00
parent a13f25400c
commit 42d02a3665
4 changed files with 10 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+1 -2
View File
@@ -83,9 +83,8 @@ def full_vae_decode(latents, model):
sd_models.move_base(model, base_device)
t1 = time.time()
if debug:
log_debug(f'VAE config: {model.vae.config}')
log_debug(f'VAE memory: {shared.mem_mon.read()}')
log_debug(f'VAE decode: name={sd_vae.loaded_vae_file if sd_vae.loaded_vae_file is not None else "baked"} dtype={model.vae.dtype} upcast={upcast} images={latents.shape[0]} latents={latents.shape} time={round(t1-t0, 3)}')
log_debug(f'VAE decode: name={sd_vae.loaded_vae_file if sd_vae.loaded_vae_file is not None else "baked"} dtype={model.vae.dtype} upcast={upcast} slicing={getattr(model.vae, "use_slicing", None)} tiling={getattr(model.vae, "use_tiling", None)} images={latents.shape[0]} latents={latents.shape} time={round(t1-t0, 3)}')
return decoded