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

pulid optimizations: dtype, vae, offload

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-09 20:03:41 -05:00
parent 79f96fb509
commit 6510a13904
7 changed files with 103 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+9 -2
View File
@@ -35,7 +35,7 @@ def create_latents(image, p, dtype=None, device=None):
def full_vae_decode(latents, model):
t0 = time.time()
if not hasattr(model, 'vae'):
if model is None or not hasattr(model, 'vae'):
shared.log.error('VAE not found in model')
return []
if debug:
@@ -170,7 +170,14 @@ def vae_decode(latents, model, output_type='np', full_quality=True, width=None,
if latents.shape[-1] <= 4: # not a latent, likely an image
decoded = latents.float().cpu().numpy()
elif full_quality and hasattr(shared.sd_model, "vae"):
decoded = full_vae_decode(latents=latents, model=shared.sd_model)
parent = shared.sd_model if hasattr(shared.sd_model, 'vae') else None
if hasattr(shared.sd_model, 'vae'):
parent = shared.sd_model
elif hasattr(shared.sd_model, 'pipe') and hasattr(shared.sd_model.pipe, 'vae'):
parent = shared.sd_model.pipe
else:
parent = None
decoded = full_vae_decode(latents=latents, model=parent)
else:
decoded = taesd_vae_decode(latents=latents)