Revert "tried to revert to fad8be1"

This reverts commit a34c7f7.

Author: eltsai

src/maxdiffusion/tests/ltx2/test_attention_ltx2.py

@@ -378,6 +378,7 @@ def test_attention_mask_parity(self):
 
     jax_model.attention_op.attention_kernel = "flash"
     jax_model.attention_op.mesh = mesh
+    jax_model.attention_op.flash_min_seq_length = 0
 
     mask_pattern_np = np.random.randint(0, 2, (self.B, S_flash)).astype(np.float32)
     pt_mask_additive = torch.from_numpy((1.0 - mask_pattern_np) * -1e9)[:, None, None, :]