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

Merge pull request #910 from ldiex/main

Browse Source 
Fix gradient checkpointing in WAN VACE blocks
This commit is contained in:
Zhongjie Duan
2025-09-23 12:23:12 +08:00
committed by GitHub
parent f6534a5b63 3a0d51d100
commit 32b8b9b51e
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: