Update pipeline.py

pipeline.py

@@ -962,12 +962,14 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
         with self.progress_bar(total=len(timesteps)) as progress_bar:
             for i, t in enumerate(timesteps):
-                
+                print(f"Step: {i}")
+                print(f"Timestep: {t}")
                 latent_sum = torch.zeros_like(latents).to(device).to(dtype=torch.float16)
                 latent_counter = torch.zeros(num_frames).to(device).to(dtype=torch.float16)
         
                 # foreach context group seperately denoise the current timestep
                 for context_group in range(num_context_groups):
+                    print(f"Context group: {context_group}")
                     # calculate to current indexes, considering overlap
                     if context_group == 0:current_context_start = 0
                     else:current_context_start = context_group * (context_size - overlap)