Update pipeline.py

pipeline.py

@@ -50,7 +50,6 @@ from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 from diffusers.utils import deprecate, is_compiled_module
 
 
-
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
 EXAMPLE_DOC_STRING = """
@@ -721,10 +720,11 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         
         # divide the initial latents into context groups
         num_context_groups = num_frames // (context_size-overlap)
+        print(f"Num context groups: {num_context_groups}")
         
         # Denoising loop
         num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
-        with self.progress_bar(total=num_frames-(num_context_groups*overlap)) as progress_bar:
+        with self.progress_bar(total=num_context_groups * len(timesteps)) as progress_bar:
             for i, t in enumerate(timesteps):
                 
                 latent_sum = torch.zeros_like(latents).to(device).to(dtype=torch.float16)