nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai
2025-12-23 17:18:55 +02:00
parent 8509236da1
commit 222fc70eb7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
LongVie2/nodes.py
+2 -2
View File
@@ -48,8 +48,8 @@ class WanVideoAddDualControlEmbeds:
vae.to(device) vae.to(device)
if dense is not None: if dense is not None:
dense_images = dense[..., :3].permute(3, 0, 1, 2) * 2 - 1 dense_images = 1 - dense[..., :3] # Invert colors for depth to match the usual range in comfy
dense_images = 1 - dense_images # Invert colors for depth to match the usual range in comfy dense_images = dense_images.permute(3, 0, 1, 2) * 2 - 1
dense_video_latent = vae.encode([dense_images.to(device, vae.dtype)], device, tiled=False) dense_video_latent = vae.encode([dense_images.to(device, vae.dtype)], device, tiled=False)
dense_first = (dense_images[:, :1]).to(device, vae.dtype) dense_first = (dense_images[:, :1]).to(device, vae.dtype)
vae_input_dense = torch.cat([dense_first, torch.zeros(3, num_frames-1, height, width, device=device, dtype=vae.dtype)], dim=1) vae_input_dense = torch.cat([dense_first, torch.zeros(3, num_frames-1, height, width, device=device, dtype=vae.dtype)], dim=1)