Update src/diffusers/models/transformers/sana_transformer_2d.py

Co-authored-by: hlky <[email protected]>

Author: lawrence-cj

src/diffusers/models/transformers/sana_transformer_2d.py

@@ -682,7 +682,9 @@ def custom_forward(*inputs):
                     encoder_attention_mask,
                     timestep,
                     cross_attention_kwargs,
-                    HW=(height, width),
+                    None,
+                    added_cond_kwargs,
+                    (height, width),
                     **ckpt_kwargs,
                 )
             else: