attn_mask correction

prishajain1 · prishajain1 · commit 00b7878e31b3 · 2026-01-10T22:42:42.000+05:30
diff --git a/src/maxdiffusion/models/wan/transformers/transformer_wan.py b/src/maxdiffusion/models/wan/transformers/transformer_wan.py
@@ -621,8 +621,7 @@ def __call__(
       def scan_fn(carry, block):
         hidden_states_carry, rngs_carry = carry
         hidden_states = block(
-            hidden_states_carry, encoder_hidden_states, timestep_proj, rotary_emb, deterministic, rngs_carry,
-            encoder_attention_mask=encoder_attention_mask
+            hidden_states_carry, encoder_hidden_states, timestep_proj, rotary_emb, deterministic, rngs_carry
         )
         new_carry = (hidden_states, rngs_carry)
         return new_carry, None
