From 6bb1677bd46a45d05583c6cda9050dd67dae4adf Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Mon, 28 Jul 2025 08:13:59 -1000 Subject: [PATCH] Update src/diffusers/models/transformers/transformer_wan.py --- 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 2c8b76912b..b6c01c13c1 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 (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)