Fix bug with WanAnimateToVideo. (#9990)

This commit is contained in:
comfyanonymous 2025-09-22 14:34:33 -07:00 committed by GitHub
parent 8a5ac527e6
commit 707b2638ec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1229,7 +1229,7 @@ class WanAnimateToVideo(io.ComfyNode):
character_mask = character_mask.unsqueeze(1)
character_mask = comfy.utils.common_upscale(character_mask[:, :, :length], concat_latent_image.shape[-1], concat_latent_image.shape[-2], "nearest-exact", "center")
if character_mask.shape[2] > ref_images_num:
mask_refmotion[:, :, ref_images_num:character_mask.shape[2] + ref_images_num] = character_mask[:, :, ref_images_num:]
mask_refmotion[:, :, ref_images_num:character_mask.shape[2]] = character_mask[:, :, ref_images_num:]
concat_latent_image = torch.cat((concat_latent_image, vae.encode(image[:, :, :, :3])), dim=2)