wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-24 17:32:40 +08:00
Code Issues Actions 32 Packages Projects Releases Wiki Activity

Add latents fix.
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Details
Python Linting / Run Pylint (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Talmaj Marinc 2026-04-10 11:11:36 +02:00
parent 2a45d487d4
commit a0a69c9b34
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_void.py
View File

@ -1,6 +1,8 @@
import nodes import nodes
import node_helpers import node_helpers
import torch import torch
import comfy
import comfy.latent_formats
import comfy.model_management import comfy.model_management
import comfy.utils import comfy.utils
from comfy_api.latest import io, ComfyExtension from comfy_api.latest import io, ComfyExtension
@ -144,6 +146,12 @@ class VOIDInpaintConditioning(io.ComfyNode):
inpaint_latents = torch.cat([mask_latents, masked_video_latents], dim=1) inpaint_latents = torch.cat([mask_latents, masked_video_latents], dim=1)
# CogVideoX.concat_cond() applies process_latent_in (x scale_factor) to
# concat_latent_image before feeding it to the transformer. Pre-divide here
# so the net scaling is identity — the VOID model expects raw VAE latents.
scale_factor = comfy.latent_formats.CogVideoX().scale_factor
inpaint_latents = inpaint_latents / scale_factor
positive = node_helpers.conditioning_set_values( positive = node_helpers.conditioning_set_values(
positive, {"concat_latent_image": inpaint_latents} positive, {"concat_latent_image": inpaint_latents}
) )