mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-12-15 19:53:22 +00:00
removed test leftover
This commit is contained in:
parent
93d99aa2bd
commit
32f1665f3b
@ -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,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user