Skip to content

Commit c0af998

Browse files
committed
Merge branch 'wan2.2' of github.com:huggingface/diffusers into wan2.2
2 parents 1ff7c99 + 6bb1677 commit c0af998

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_wan.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ def forward(
325325
c_scale_msa = c_scale_msa.squeeze(2)
326326
c_gate_msa = c_gate_msa.squeeze(2)
327327
else:
328-
# temb: batch_size, 6, inner_dim (wan2.1)
328+
# temb: batch_size, 6, inner_dim (wan2.1/wan2.2 14B)
329329
shift_msa, scale_msa, gate_msa, c_shift_msa, c_scale_msa, c_gate_msa = (
330330
self.scale_shift_table + temb.float()
331331
).chunk(6, dim=1)

0 commit comments

Comments
 (0)