Update pipeline.py

pipeline.py

@@ -1497,9 +1497,14 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         
                     if do_classifier_free_guidance:
                         # Split tensor along its first dimension
+                        torch.cuda.synchronize()  # Synchronize GPU before starting timing
                         time_chunk = time.time()
+
                         noise_pred_uncond, noise_pred_text = torch.chunk(noise_pred, 2, dim=0)
+
+                        torch.cuda.synchronize()  # Synchronize GPU after operation
                         print("chunk time", time.time() - time_chunk)
+                        
                         for i, context_index in enumerate(current_context_indexes):
                             # Perform batch addition
                             noise_pred_uncond_sum[..., context_index, :, :] += noise_pred_uncond[:, :, i, :, :]