mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-12-16 20:23:21 +00:00
removed test leftover
This commit is contained in:
parent
93d99aa2bd
commit
32f1665f3b
@ -368,8 +368,8 @@ class WanT2V:
|
|||||||
shift=1,
|
shift=1,
|
||||||
use_dynamic_shifting=False)
|
use_dynamic_shifting=False)
|
||||||
sample_scheduler.set_timesteps(
|
sample_scheduler.set_timesteps(
|
||||||
sampling_steps + 4, device=self.device, shift=shift)
|
sampling_steps, device=self.device, shift=shift)
|
||||||
timesteps = sample_scheduler.timesteps[:-4]
|
timesteps = sample_scheduler.timesteps
|
||||||
elif sample_solver == 'dpm++':
|
elif sample_solver == 'dpm++':
|
||||||
sample_scheduler = FlowDPMSolverMultistepScheduler(
|
sample_scheduler = FlowDPMSolverMultistepScheduler(
|
||||||
num_train_timesteps=self.num_train_timesteps,
|
num_train_timesteps=self.num_train_timesteps,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user