xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 23:57:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

save

Browse Source
This commit is contained in:
Eugene Fairley 2025-06-27 20:54:44 -07:00
parent 78c52aac5f
commit 9707f9d4b0
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/model_base.py
View File

@ -1076,10 +1076,11 @@ class WAN21(BaseModel):
shape_image[1] = extra_channels
image = torch.zeros(shape_image, dtype=noise.dtype, layout=noise.layout, device=noise.device)
else:
image = utils.common_upscale(image.to(device), noise.shape[-1], noise.shape[-2], "bilinear", "center")
for i in range(0, image.shape[1], 36):
image[:, i: i + 36] = self.process_latent_in(image[:, i: i + 36])
image = utils.resize_to_batch_size(image, noise.shape[0])
image = image.to(device)
# image = utils.common_upscale(image.to(device), noise.shape[-1], noise.shape[-2], "bilinear", "center")
# for i in range(0, image.shape[1], 36):
# image[:, i: i + 36] = self.process_latent_in(image[:, i: i + 36])
# image = utils.resize_to_batch_size(image, noise.shape[0])
print(f"image shape: {image.shape}")
if not self.image_to_video or extra_channels == image.shape[1]: