mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
taesd preview modularize
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user