support ltx2 gradient_checkpointing

2026-03-22 00:38:11 +00:00 · 2026-02-26 19:19:59 +08:00
parent a87910bc65
commit a18966c300
10 changed files with 36 additions and 33 deletions
--- a/diffsynth/pipelines/ltx2_audio_video.py
+++ b/diffsynth/pipelines/ltx2_audio_video.py
@@ -577,6 +577,8 @@ def model_fn_ltx2(
        audio_positions=audio_positions,
        audio_context=audio_context,
        audio_timesteps=audio_timesteps,
+        use_gradient_checkpointing=use_gradient_checkpointing,
+        use_gradient_checkpointing_offload=use_gradient_checkpointing_offload,
    )
    # unpatchify
    vx = video_patchifier.unpatchify_video(vx, f, h, w)