Apply suggestions from code review

Co-authored-by: bagheera <59658056+bghira@users.noreply.github.com>

Author: yiyixuxu

src/diffusers/models/transformers/transformer_wan.py

@@ -325,7 +325,7 @@ def forward(
             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)