Skip to content

Commit a34c7f7

Browse files
committed
tried to revert to fad8be1
1 parent 2e9f896 commit a34c7f7

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

src/maxdiffusion/tests/ltx2/test_attention_ltx2.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,6 @@ def test_attention_mask_parity(self):
378378

379379
jax_model.attention_op.attention_kernel = "flash"
380380
jax_model.attention_op.mesh = mesh
381-
jax_model.attention_op.flash_min_seq_length = 0
382381

383382
mask_pattern_np = np.random.randint(0, 2, (self.B, S_flash)).astype(np.float32)
384383
pt_mask_additive = torch.from_numpy((1.0 - mask_pattern_np) * -1e9)[:, None, None, :]

0 commit comments

Comments
 (0)