From 97675c703679584635d08dc29b4ecdfddf57c0a7 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Mon, 28 Jul 2025 08:13:16 -1000 Subject: [PATCH] Apply suggestions from code review Co-authored-by: bagheera <59658056+bghira@users.noreply.github.com> --- src/diffusers/models/transformers/transformer_wan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/transformers/transformer_wan.py b/src/diffusers/models/transformers/transformer_wan.py index eddf196718..2c8b76912b 100644 --- a/src/diffusers/models/transformers/transformer_wan.py +++ b/src/diffusers/models/transformers/transformer_wan.py @@ -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)