From d47ac9cbe06d3f30e8c4ea16516865c52e87b635 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 13 Sep 2025 19:43:12 +0300 Subject: [PATCH] Update nodes.py --- HuMo/nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/HuMo/nodes.py b/HuMo/nodes.py index ec01f9d..507c554 100644 --- a/HuMo/nodes.py +++ b/HuMo/nodes.py @@ -212,13 +212,13 @@ class HuMoEmbeds: reference_images_in = common_upscale(reference_images.movedim(-1, 1), width, height, "lanczos", "disabled").movedim(-1, 1) else: reference_images_in = reference_images - samples, = WanVideoEncodeLatentBatch.encode(self, vae, reference_images_in, False, 0, 0, 0, 0) + samples, = WanVideoEncodeLatentBatch.encode(self, vae, reference_images_in, tiled_vae, 0, 0, 0, 0) samples = samples["samples"].transpose(0, 2).squeeze(0) num_refs = samples.shape[1] vae.to(device) zero_frames = torch.zeros(1, 3, pixel_frame_num + 4*num_refs, height, width, device=device, dtype=vae.dtype) - zero_latents = vae.encode(zero_frames, device=device, tiled_vae=tiled_vae)[0].to(offload_device) + zero_latents = vae.encode(zero_frames, device=device, tiled=tiled_vae)[0].to(offload_device) vae.model.clear_cache() vae.to(offload_device) mm.soft_empty_cache()