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

Fix gradient checkpointing in WAN VACE blocks

Browse Source
This commit is contained in:
Tianlin Pan
2025-09-14 16:21:46 +08:00
committed by GitHub
parent d93e8738cd
commit 3a0d51d100
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
diffsynth/pipelines/wan_video_new.py
View File

@@ -1243,7 +1243,11 @@ def model_fn_wan_video(
tea_cache_update = False
if vace_context is not None:
vace_hints = vace(x, vace_context, context, t_mod, freqs)
vace_hints = vace(
x, vace_context, context, t_mod, freqs,
use_gradient_checkpointing=use_gradient_checkpointing,
use_gradient_checkpointing_offload=use_gradient_checkpointing_offload
)
# blocks
if use_unified_sequence_parallel: