diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py index 550513b5c943..1ede718f21b7 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py @@ -126,7 +126,7 @@ def __call__( if accepts_eta: extra_step_kwargs["eta"] = eta - for i, t in tqdm(enumerate(self.scheduler.timesteps)): + for i, t in tqdm(enumerate(self.scheduler.timesteps), leave=False): # expand the latents if we are doing classifier free guidance latent_model_input = torch.cat([latents] * 2) if do_classifier_free_guidance else latents if isinstance(self.scheduler, LMSDiscreteScheduler):