support ltx2 gradient_checkpointing

2026-03-23 17:38:10 +00:00 · 2026-02-26 19:19:59 +08:00
parent a87910bc65
commit a18966c300
10 changed files with 36 additions and 33 deletions
--- a/examples/ltx2/model_training/train.py
+++ b/examples/ltx2/model_training/train.py
@@ -118,10 +118,10 @@ if __name__ == "__main__":
            max_pixels=args.max_pixels,
            height=args.height,
            width=args.width,
-            height_division_factor=16,
-            width_division_factor=16,
+            height_division_factor=32,
+            width_division_factor=32,
            num_frames=args.num_frames,
-            time_division_factor=4,
+            time_division_factor=8,
            time_division_remainder=1,
        ),
        special_operator_map={