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

Apply suggestions from code review

Browse Source 
Co-authored-by: bagheera <59658056+bghira@users.noreply.github.com>
This commit is contained in:
YiYi Xu
2025-07-28 08:13:16 -10:00
committed by GitHub
parent e78a4fa30a
commit 97675c7036
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
# temb: batch_size, 6, inner_dim (wan2.1)
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)