From 32f1665f3b541d26027e0b3961e291e402c341a3 Mon Sep 17 00:00:00 2001 From: deepbeepmeep Date: Sat, 26 Apr 2025 15:49:08 +0200 Subject: [PATCH] removed test leftover --- wan/text2video.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wan/text2video.py b/wan/text2video.py index ca5617b..45f4149 100644 --- a/wan/text2video.py +++ b/wan/text2video.py @@ -368,8 +368,8 @@ class WanT2V: shift=1, use_dynamic_shifting=False) sample_scheduler.set_timesteps( - sampling_steps + 4, device=self.device, shift=shift) - timesteps = sample_scheduler.timesteps[:-4] + sampling_steps, device=self.device, shift=shift) + timesteps = sample_scheduler.timesteps elif sample_solver == 'dpm++': sample_scheduler = FlowDPMSolverMultistepScheduler( num_train_timesteps=self.num_train_timesteps,