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

fix pulid vae

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-11 10:46:33 -05:00
parent 15381eb103
commit 70eec4349e
1 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+3 -9
View File
@@ -147,6 +147,7 @@ def taesd_vae_encode(image):
def vae_decode(latents, model, output_type='np', full_quality=True, width=None, height=None):
t0 = time.time()
model = model or shared.sd_model
if latents is None or not torch.is_tensor(latents): # already decoded
return latents
prev_job = shared.state.job
@@ -169,15 +170,8 @@ 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"):
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)
elif full_quality and hasattr(model, "vae"):
decoded = full_vae_decode(latents=latents, model=model)
else:
decoded = taesd_vae_decode(latents=latents)