smoothieAI commited on
Commit
2b630ab
·
verified ·
1 Parent(s): 962fb36

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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=num_context_groups * 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)
 
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)