Skip to content

GitLab

Commit aaf5f643 authored by sandy's avatar sandy Committed by GitHub
Browse files

fix: video stitching flicker/shrink issue (#263)

parent 74e443ef
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
lightx2v/models/runners/wan/wan_audio_runner.py
View file @ aaf5f643
...@@ -362,7 +362,8 @@ class WanAudioRunner(WanRunner): # type:ignore ...@@ -362,7 +362,8 @@ class WanAudioRunner(WanRunner): # type:ignore
frames_n = (nframe - 1) * 4 + 1 frames_n = (nframe - 1) * 4 + 1
prev_mask = torch.ones((1, frames_n, height, width), device=device, dtype=dtype) prev_mask = torch.ones((1, frames_n, height, width), device=device, dtype=dtype)
prev_mask[:, prev_len:] = 0 prev_frame_len = max((prev_len - 1) * 4 + 1, 0)
prev_mask[:, prev_frame_len:] = 0
prev_mask = self._wan_mask_rearrange(prev_mask) prev_mask = self._wan_mask_rearrange(prev_mask)
if prev_latents is not None: if prev_latents is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment