xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-09 04:44:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix logger

Browse Source
This commit is contained in:
kijai 2024-10-21 09:45:23 +03:00
parent 01bce7dfff
commit dc4bb73d3f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pipeline_cogvideox.py
View File

@ -516,7 +516,7 @@ class CogVideoXPipeline(VideoSysPipeline):
latent_padding = torch.zeros(padding_shape, device=device, dtype=self.vae.dtype) latent_padding = torch.zeros(padding_shape, device=device, dtype=self.vae.dtype)
image_cond_latents = torch.cat([image_cond_latents[:, 0, :, :, :].unsqueeze(1), latent_padding, image_cond_latents[:, -1, :, :, :].unsqueeze(1)], dim=1) image_cond_latents = torch.cat([image_cond_latents[:, 0, :, :, :].unsqueeze(1), latent_padding, image_cond_latents[:, -1, :, :, :].unsqueeze(1)], dim=1)
logger.info("image cond latents shape: ",image_cond_latents.shape) logger.info(f"image cond latents shape: {image_cond_latents.shape}")
else: else:
logger.info("Only one image conditioning frame received, img2vid") logger.info("Only one image conditioning frame received, img2vid")
padding_shape = ( padding_shape = (
@ -542,7 +542,7 @@ class CogVideoXPipeline(VideoSysPipeline):
# masks # masks
if self.original_mask is not None: if self.original_mask is not None:
mask = self.original_mask.to(device) mask = self.original_mask.to(device)
logger.info("self.original_mask: ", self.original_mask.shape) logger.info(f"self.original_mask: {self.original_mask.shape}")
mask = F.interpolate(self.original_mask.unsqueeze(1), size=(latents.shape[-2], latents.shape[-1]), mode='bilinear', align_corners=False) mask = F.interpolate(self.original_mask.unsqueeze(1), size=(latents.shape[-2], latents.shape[-1]), mode='bilinear', align_corners=False)
if mask.shape[0] != latents.shape[1]: if mask.shape[0] != latents.shape[1]: