298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't error if wan concat image has extra channels.

Browse Source
This commit is contained in:
comfyanonymous 2025-03-28 08:40:25 -04:00
parent a40fcfc2d5
commit 2d17d8910c
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_base.py
View File

@ -1013,6 +1013,9 @@ class WAN21(BaseModel):
if not self.image_to_video or extra_channels == image.shape[1]: if not self.image_to_video or extra_channels == image.shape[1]:
return image return image
if image.shape[1] > (extra_channels - 4):
image = image[:, :(extra_channels - 4)]
mask = kwargs.get("concat_mask", kwargs.get("denoise_mask", None)) mask = kwargs.get("concat_mask", kwargs.get("denoise_mask", None))
if mask is None: if mask is None:
mask = torch.zeros_like(noise)[:, :4] mask = torch.zeros_like(noise)[:, :4]