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

fix a bug in sliding_window inference

Browse Source
This commit is contained in:
ziyannchen
2025-07-16 10:30:33 +00:00
parent d19fcc8c04
commit 05c6b49b90
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/pipelines/wan_video_new.py
View File

@@ -1047,7 +1047,7 @@ class TemporalTiler_BCTHW:
mask = self.build_mask( mask = self.build_mask(
model_output, model_output,
is_bound=(t == 0, t_ == T), is_bound=(t == 0, t_ == T),
border_width=(sliding_window_size - sliding_window_stride,) border_width=(sliding_window_size - sliding_window_stride + 1,)
).to(device=data_device, dtype=data_dtype) ).to(device=data_device, dtype=data_dtype)
value[:, :, t: t_, :, :] += model_output * mask value[:, :, t: t_, :, :] += model_output * mask
weight[:, :, t: t_, :, :] += mask weight[:, :, t: t_, :, :] += mask