taesd preview modularize

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-05-09 10:15:26 +02:00
parent 4747ca2fc0
commit 12d5ce65ee
2 changed files with 25 additions and 13 deletions
+1 -1
View File
@@ -45,7 +45,7 @@ def vae_decode_tiny(latents):
global taesd # pylint: disable=global-statement
if taesd is None:
from modules.vae import sd_vae_taesd
taesd, _variant = sd_vae_taesd.get_model(variant='TAE HunyuanVideo')
taesd, _variant = sd_vae_taesd.load_model(variant='TAE HunyuanVideo')
log.debug(f'Video VAE: type=Tiny cls={taesd.__class__.__name__} latents={latents.shape}')
with devices.inference_context():
taesd = taesd.to(device=devices.device, dtype=devices.dtype)