From b3ef224042007698278d77a481071f6cd56aedbf Mon Sep 17 00:00:00 2001
From: Zhongjie Duan <35051019+Artiprocher@users.noreply.github.com>
Date: Mon, 2 Mar 2026 19:06:55 +0800
Subject: [PATCH] support Anima gradient checkpointing (#1319)

---
 diffsynth/pipelines/anima_image.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/diffsynth/pipelines/anima_image.py b/diffsynth/pipelines/anima_image.py
index 732ede5..32a3c71 100644
--- a/diffsynth/pipelines/anima_image.py
+++ b/diffsynth/pipelines/anima_image.py
@@ -256,6 +256,8 @@ def model_fn_anima(
         timesteps=timestep,
         context=prompt_emb,
         t5xxl_ids=t5xxl_ids,
+        use_gradient_checkpointing=use_gradient_checkpointing,
+        use_gradient_checkpointing_offload=use_gradient_checkpointing_offload,
     )
     model_output = model_output.squeeze(2)
     return model_output