Skip to content

Commit 1ce172b

Browse files
committed
feat: enable FSDP in LTX2 smoke test to avoid OOM in CI
1 parent b931522 commit 1ce172b

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/maxdiffusion/tests/ltx2_smoke_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,10 @@ def setUpClass(cls):
4848
"num_frames=9", # Small number of frames
4949
"seed=0",
5050
"attention=flash",
51+
"ici_fsdp_parallelism=8",
52+
"ici_data_parallelism=1",
53+
"ici_context_parallelism=1",
54+
"ici_tensor_parallelism=1",
5155
],
5256
unittest=True,
5357
)

0 commit comments

Comments
 (0)