Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -715,7 +715,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
715 |
|
716 |
# Denoising loop
|
717 |
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
|
718 |
-
with self.progress_bar(total=
|
719 |
for i, t in enumerate(timesteps):
|
720 |
|
721 |
latent_sum = torch.zeros_like(latents).to(device).to(dtype=torch.float16)
|
|
|
715 |
|
716 |
# Denoising loop
|
717 |
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
|
718 |
+
with self.progress_bar(total=len(timesteps)) as progress_bar:
|
719 |
for i, t in enumerate(timesteps):
|
720 |
|
721 |
latent_sum = torch.zeros_like(latents).to(device).to(dtype=torch.float16)
|