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-11-10 21:02:58 +02:00
parent e4a4d22537
commit 907c9e1cdd
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+5 -6
View File
@@ -812,16 +812,15 @@ class WanVideoAddBindweaveEmbeds:
clip_embeds = updated.get("clip_context", None) clip_embeds = updated.get("clip_context", None)
if clip_embeds is not None: if clip_embeds is not None:
if clip_embeds.shape[1] != 257:
clip_embeds = clip_embeds.view(clip_embeds.shape[1] // 257, 257, clip_embeds.shape[2])
N = clip_embeds.shape[0]
B, T, C = clip_embeds.shape B, T, C = clip_embeds.shape
pad = torch.zeros(B, 0, C, device=clip_embeds.device, dtype=clip_embeds.dtype) target_len = max_refs * 257 # 4 * 257 = 1028
if N < max_refs: if T < target_len:
pad = torch.zeros(B, (max_refs-N)*T, C, device=clip_embeds.device, dtype=clip_embeds.dtype) pad = torch.zeros(B, target_len - T, C, device=clip_embeds.device, dtype=clip_embeds.dtype)
padded_embeds = torch.cat([clip_embeds, pad], dim=1) padded_embeds = torch.cat([clip_embeds, pad], dim=1)
log.info(f"Padded clip embeds from {clip_embeds.shape} to {padded_embeds.shape} for Bindweave") log.info(f"Padded clip embeds from {clip_embeds.shape} to {padded_embeds.shape} for Bindweave")
updated["clip_context"] = padded_embeds updated["clip_context"] = padded_embeds
else:
updated["clip_context"] = clip_embeds
updated["image_embeds"] = image_embeds updated["image_embeds"] = image_embeds
updated["qwenvl_embeds_pos"] = qwenvl_embeds_pos updated["qwenvl_embeds_pos"] = qwenvl_embeds_pos