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 control batch and hires

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-17 14:53:15 -04:00
parent 2508f4f245
commit d0e35a7a6b
4 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+2 -2
View File
@@ -120,10 +120,10 @@ def vae_decode(latents, model, output_type='np', full_quality=True):
if not hasattr(model, 'vae'):
shared.log.error('VAE not found in model')
return []
if latents.shape[0] == 4 and latents.shape[1] != 4: # likely animatediff latent
latents = latents.permute(1, 0, 2, 3)
if len(latents.shape) == 3: # lost a batch dim in hires
latents = latents.unsqueeze(0)
if latents.shape[0] == 4 and latents.shape[1] != 4: # likely animatediff latent
latents = latents.permute(1, 0, 2, 3)
if full_quality:
decoded = full_vae_decode(latents=latents, model=shared.sd_model)
else: