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