jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Slightly better wan i2v mask implementation.

Browse Source
This commit is contained in:
comfyanonymous
2025-02-26 03:49:50 -05:00
parent 4ced06b879
commit 0844998db3
2 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
comfy_extras/nodes_wan.py
View File

@@ -5,13 +5,6 @@ import comfy.model_management
import comfy.utils
def masked_images(num_images):
rem = 4 - (num_images % 4)
if rem == 4:
return num_images
return rem + num_images
class WanImageToVideo:
@classmethod
def INPUT_TYPES(s):
@@ -41,8 +34,8 @@ class WanImageToVideo:
image[:start_image.shape[0]] = start_image
concat_latent_image = vae.encode(image[:, :, :, :3])
mask = torch.ones((1, 1, latent.shape[2] * 4, concat_latent_image.shape[-2], concat_latent_image.shape[-1]), device=start_image.device, dtype=start_image.dtype)
mask[:, :, :masked_images(start_image.shape[0])] = 0.0
mask = torch.ones((1, 1, latent.shape[2], concat_latent_image.shape[-2], concat_latent_image.shape[-1]), device=start_image.device, dtype=start_image.dtype)
mask[:, :, :((start_image.shape[0] - 1) // 4) + 1] = 0.0
positive = node_helpers.conditioning_set_values(positive, {"concat_latent_image": concat_latent_image, "concat_mask": mask})
negative = node_helpers.conditioning_set_values(negative, {"concat_latent_image": concat_latent_image, "concat_mask": mask})