Skip to content

Commit a0a69c9

Browse files
author
Talmaj Marinc
committed
Add latents fix.
1 parent 2a45d48 commit a0a69c9

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

comfy_extras/nodes_void.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import nodes
22
import node_helpers
33
import torch
4+
import comfy
5+
import comfy.latent_formats
46
import comfy.model_management
57
import comfy.utils
68
from comfy_api.latest import io, ComfyExtension
@@ -144,6 +146,12 @@ def _match_temporal(lat, target_t):
144146

145147
inpaint_latents = torch.cat([mask_latents, masked_video_latents], dim=1)
146148

149+
# CogVideoX.concat_cond() applies process_latent_in (x scale_factor) to
150+
# concat_latent_image before feeding it to the transformer. Pre-divide here
151+
# so the net scaling is identity — the VOID model expects raw VAE latents.
152+
scale_factor = comfy.latent_formats.CogVideoX().scale_factor
153+
inpaint_latents = inpaint_latents / scale_factor
154+
147155
positive = node_helpers.conditioning_set_values(
148156
positive, {"concat_latent_image": inpaint_latents}
149157
)

0 commit comments

Comments
 (0)