theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

support Anima gradient checkpointing (#1319)

Browse Source
This commit is contained in:
Zhongjie Duan
2026-03-02 19:06:55 +08:00
committed by GitHub
parent f43b18ec21
commit b3ef224042
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/pipelines/anima_image.py
View File

@@ -256,6 +256,8 @@ def model_fn_anima(
timesteps=timestep, timesteps=timestep,
context=prompt_emb, context=prompt_emb,
t5xxl_ids=t5xxl_ids, t5xxl_ids=t5xxl_ids,
use_gradient_checkpointing=use_gradient_checkpointing,
use_gradient_checkpointing_offload=use_gradient_checkpointing_offload,
) )
model_output = model_output.squeeze(2) model_output = model_output.squeeze(2)
return model_output return model_output