AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Merge branch 'wan2.2' of github.com:huggingface/diffusers into wan2.2

Browse Source
This commit is contained in:
yiyixuxu
2025-07-28 21:48:47 +02:00
parent 1ff7c99596 6bb1677bd4
commit c0af998473
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/transformers/transformer_wan.py
View File

@@ -325,7 +325,7 @@ class WanTransformerBlock(nn.Module):
c_scale_msa = c_scale_msa.squeeze(2)
c_gate_msa = c_gate_msa.squeeze(2)
else:
# temb: batch_size, 6, inner_dim (wan2.1)
# temb: batch_size, 6, inner_dim (wan2.1/wan2.2 14B)
shift_msa, scale_msa, gate_msa, c_shift_msa, c_scale_msa, c_gate_msa = (
self.scale_shift_table + temb.float()
).chunk(6, dim=1)