AI/ComfyUI-WanVideoWrapper
1
0
Fork 0
mirror of https://github.com/kijai/ComfyUI-WanVideoWrapper.git synced 2026-01-28 12:20:55 +03:00
Code Activity

allow stand-in to work with chunked rope

Browse Source
This commit is contained in:
kijai
2025-08-14 14:12:06 +03:00
parent b29756a962
commit f3d5f6b3ab
2 changed files with 533 additions and 589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1111
example_workflows/wanvideo_Stand-In_reference_example_01.json
View File

File diff suppressed because it is too large Load Diff

11
wanvideo/modules/model.py
View File

@@ -774,11 +774,16 @@ class WanAttentionBlock(nn.Module):
x_main, x_ip_input = input_x[:, : -self.self_attn.cond_size], input_x[:, -self.self_attn.cond_size :]
# Compute QKV for main content
q, k, v = self.self_attn.qkv_fn(x_main)
q, k = apply_rope_comfy(q, k, freqs)
if self.rope_func == "comfy":
q, k = apply_rope_comfy(q, k, freqs)
elif self.rope_func == "comfy_chunked":
q, k = apply_rope_comfy_chunked(q, k, freqs)
# Compute QKV for IP content
q_ip, k_ip, v_ip = self.self_attn.qkv_fn_ip(x_ip_input)
q_ip, k_ip = apply_rope_comfy(q_ip, k_ip, freqs_ip)
if self.rope_func == "comfy":
q_ip, k_ip = apply_rope_comfy(q_ip, k_ip, freqs_ip)
elif self.rope_func == "comfy_chunked":
q_ip, k_ip = apply_rope_comfy_chunked(q_ip, k_ip, freqs_ip)
else:
q, k, v = self.self_attn.qkv_fn(input_x)
if self.rope_func == "comfy":