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

fix multitalk

Browse Source
This commit is contained in:
kijai
2025-08-17 18:24:48 +03:00
parent 1d7e0848e0
commit 427e7be6c2
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
multitalk/multitalk.py
+5 -4
View File
@@ -256,12 +256,13 @@ class SingleStreamAttention(nn.Module):
N_t, N_h, N_w = shape
x_extra = None
if x.shape[0] != encoder_hidden_states.shape[0]:
try:
x = rearrange(x, "B (N_t S) C -> (B N_t) S C", N_t=N_t)
except:
x_extra = x[:, -N_h * N_w:, :]
x = x[:, :-N_h * N_w, :]
N_t = N_t - 1
x = rearrange(x, "B (N_t S) C -> (B N_t) S C", N_t=N_t)
x = rearrange(x, "B (N_t S) C -> (B N_t) S C", N_t=N_t)
# get q for hidden_state
B, N, C = x.shape
@@ -367,7 +368,7 @@ class SingleStreamMultiAttention(SingleStreamAttention):
if human_num is None or human_num <= 1:
return super().forward(x, encoder_hidden_states, shape)
N_t, _, _ = shape
N_t, N_h, N_w = shape
x = rearrange(x, "B (N_t S) C -> (B N_t) S C", N_t=N_t)
x_extra = None