Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-16 12:13:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed fps broken with upsampling and window sliding

Browse Source
This commit is contained in:
DeepBeepMeep 2025-05-04 18:18:31 +02:00
parent 33e15f35ef
commit f0099e0e4b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
wgp.py
View File

@ -2891,7 +2891,7 @@ def generate_video(
exp = 1 exp = 1
elif temporal_upsampling == "rife4": elif temporal_upsampling == "rife4":
exp = 2 exp = 2
output_fps = fps
if exp > 0: if exp > 0:
from rife.inference import temporal_interpolation from rife.inference import temporal_interpolation
if sliding_window and window_no > 1: if sliding_window and window_no > 1:
@ -2901,7 +2901,7 @@ def generate_video(
else: else:
sample = temporal_interpolation( os.path.join("ckpts", "flownet.pkl"), sample, exp, device=processing_device) sample = temporal_interpolation( os.path.join("ckpts", "flownet.pkl"), sample, exp, device=processing_device)
fps = fps * 2**exp output_fps = output_fps * 2**exp
if len(spatial_upsampling) > 0: if len(spatial_upsampling) > 0:
from wan.utils.utils import resize_lanczos # need multithreading or to do lanczos with cuda from wan.utils.utils import resize_lanczos # need multithreading or to do lanczos with cuda
@ -2933,10 +2933,10 @@ def generate_video(
frames_already_processed = sample frames_already_processed = sample
if audio_guide == None: if audio_guide == None:
cache_video( tensor=sample[None], save_file=video_path, fps=fps, nrow=1, normalize=True, value_range=(-1, 1)) cache_video( tensor=sample[None], save_file=video_path, fps=output_fps, nrow=1, normalize=True, value_range=(-1, 1))
else: else:
save_path_tmp = video_path[:-4] + "_tmp.mp4" save_path_tmp = video_path[:-4] + "_tmp.mp4"
cache_video( tensor=sample[None], save_file=save_path_tmp, fps=fps, nrow=1, normalize=True, value_range=(-1, 1)) cache_video( tensor=sample[None], save_file=save_path_tmp, fps=output_fps, nrow=1, normalize=True, value_range=(-1, 1))
final_command = [ "ffmpeg", "-y", "-i", save_path_tmp, "-i", audio_guide, "-c:v", "libx264", "-c:a", "aac", "-shortest", "-loglevel", "warning", "-nostats", video_path, ] final_command = [ "ffmpeg", "-y", "-i", save_path_tmp, "-i", audio_guide, "-c:v", "libx264", "-c:a", "aac", "-shortest", "-loglevel", "warning", "-nostats", video_path, ]
import subprocess import subprocess
subprocess.run(final_command, check=True) subprocess.run(final_command, check=True)