removed test leftover

This commit is contained in:
deepbeepmeep 2025-04-26 15:49:08 +02:00
parent 93d99aa2bd
commit 32f1665f3b

View File

@ -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,