removed test leftover
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user