From ee92524b47debba2bba0a3f21b7e80794621a41f Mon Sep 17 00:00:00 2001 From: Jinay Jain Date: Wed, 28 Feb 2024 22:22:49 -0500 Subject: [PATCH] [docs] Fix step numbering in StableVideoDiffusionPipeline --- .../pipeline_stable_video_diffusion.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py index f53ebbafee2e..2989b0d754c8 100644 --- a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +++ b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py @@ -471,11 +471,11 @@ def __call__( ) added_time_ids = added_time_ids.to(device) - # 4. Prepare timesteps + # 6. Prepare timesteps self.scheduler.set_timesteps(num_inference_steps, device=device) timesteps = self.scheduler.timesteps - # 5. Prepare latent variables + # 7. Prepare latent variables num_channels_latents = self.unet.config.in_channels latents = self.prepare_latents( batch_size * num_videos_per_prompt, @@ -489,7 +489,7 @@ def __call__( latents, ) - # 7. Prepare guidance scale + # 8. Prepare guidance scale guidance_scale = torch.linspace(min_guidance_scale, max_guidance_scale, num_frames).unsqueeze(0) guidance_scale = guidance_scale.to(device, latents.dtype) guidance_scale = guidance_scale.repeat(batch_size * num_videos_per_prompt, 1) @@ -497,7 +497,7 @@ def __call__( self._guidance_scale = guidance_scale - # 8. Denoising loop + # 9. Denoising loop num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order self._num_timesteps = len(timesteps) with self.progress_bar(total=num_inference_steps) as progress_bar: