xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-07 17:01:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
Jukka Seppänen 2024-11-22 22:38:51 +02:00 committed by GitHub
parent 895d3b83a4
commit 6c7068b5bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@ -361,7 +361,7 @@ class CogVideoImageEncodeFunInP:
mask = torch.zeros_like(masked_image_latents[:, :, :1, :, :]) mask = torch.zeros_like(masked_image_latents[:, :, :1, :, :])
if end_image is not None: if end_image is not None:
mask[:, -1, :, :, :] = vae_scaling_factor mask[:, -1, :, :, :] = 0
mask[:, 0, :, :, :] = vae_scaling_factor mask[:, 0, :, :, :] = vae_scaling_factor
final_latents = masked_image_latents * vae_scaling_factor final_latents = masked_image_latents * vae_scaling_factor
@ -971,4 +971,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"CogVideoLatentPreview": "CogVideo LatentPreview", "CogVideoLatentPreview": "CogVideo LatentPreview",
"CogVideoXTorchCompileSettings": "CogVideo TorchCompileSettings", "CogVideoXTorchCompileSettings": "CogVideo TorchCompileSettings",
"CogVideoImageEncodeFunInP": "CogVideo ImageEncode FunInP", "CogVideoImageEncodeFunInP": "CogVideo ImageEncode FunInP",
} }