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 video vae selector

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-04-25 21:05:56 -04:00
parent 8c37607b82
commit cfcfa4e4d2
6 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/video_models/video_vae.py
+2 -2
View File
@@ -36,10 +36,10 @@ def vae_decode_tiny(latents):
shared.log.warning(f'Video VAE: type=Tiny cls={shared.sd_model.__class__.__name__} not supported')
return None
from modules import sd_vae_taesd
vae = sd_vae_taesd.get_model(variant)
vae = sd_vae_taesd.get_model(variant=variant)
if vae is None:
return None
debug(f'Video VAE: type=Tiny cls={vae.__class__.__name__} variant="{variant}" latents={latents.shape}')
shared.log.debug(f'Video VAE: type=Tiny cls={vae.__class__.__name__} variant="{variant}" latents={latents.shape}')
vae = vae.to(device=devices.device, dtype=devices.dtype)
latents = latents.transpose(1, 2).to(device=devices.device, dtype=devices.dtype)
images = vae.decode_video(latents, parallel=False).transpose(1, 2).mul_(2).sub_(1)