reset scheduler begin_index after using denoise on another sampler

2024-12-09 14:12:31 +02:00
parent 9a4abbc535
commit fb00591ced
1 changed files with 2 additions and 0 deletions
@@ -467,6 +467,8 @@ class HunyuanVideoPipeline(DiffusionPipeline):
        extra_set_timesteps_kwargs = self.prepare_extra_func_kwargs(
            self.scheduler.set_timesteps, {"n_tokens": n_tokens}
        )
+        if hasattr(self.scheduler, "set_begin_index") and denoise_strength == 1.0:
+            self.scheduler.set_begin_index(begin_index=0)
        timesteps, num_inference_steps = retrieve_timesteps(
            self.scheduler,
            num_inference_steps,