xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-03 04:37:06 +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 shape_image[1] = extra_channels
image = torch.zeros(shape_image, dtype=noise.dtype, layout=noise.layout, device=noise.device) image = torch.zeros(shape_image, dtype=noise.dtype, layout=noise.layout, device=noise.device)
else: else:
image = utils.common_upscale(image.to(device), noise.shape[-1], noise.shape[-2], "bilinear", "center") image = image.to(device)
for i in range(0, image.shape[1], 36): # image = utils.common_upscale(image.to(device), noise.shape[-1], noise.shape[-2], "bilinear", "center")
image[:, i: i + 36] = self.process_latent_in(image[:, i: i + 36]) # for i in range(0, image.shape[1], 36):
image = utils.resize_to_batch_size(image, noise.shape[0]) # 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}") print(f"image shape: {image.shape}")
if not self.image_to_video or extra_channels == image.shape[1]: if not self.image_to_video or extra_channels == image.shape[1]: