From fb00591cedc963cf2798bf7557c4af747d1e8053 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 9 Dec 2024 14:12:31 +0200 Subject: [PATCH] reset scheduler begin_index after using denoise on another sampler --- hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py index 51c08d5..de41e82 100644 --- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py +++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py @@ -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,