Spaces:

jbilcke-hf
/

Wan2GP_you_must_clone_this_space_to_use_it

Paused

App Files Files Community

deepbeepmeep commited on Apr 26

Commit

46bd575

·

1 Parent(s): 512bc0e

removed test leftover

Files changed (1) hide show

wan/text2video.py +2 -2

wan/text2video.py CHANGED Viewed

@@ -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]
         elif sample_solver == 'dpm++':
             sample_scheduler = FlowDPMSolverMultistepScheduler(
                 num_train_timesteps=self.num_train_timesteps,

                 shift=1,
                 use_dynamic_shifting=False)
             sample_scheduler.set_timesteps(
+                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,