Update pipeline.py

pipeline.py

@@ -783,8 +783,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 video_tensor = self.decode_latents(latents[:, :, start_idx:end_idx, :, :])
                 video = tensor2vid(video_tensor, self.image_processor, output_type=output_type)
 
-                for batch_idx, frame_batch in enumerate(video):
-                    for frame_idx, frame in enumerate(frame_batch[0]):
+                for batch_idx, frame_batch in enumerate(video[0][0]):
+                    for frame_idx, frame in enumerate(frame_batch):
                         frame_number = start_idx + batch_idx * len(frame_batch[0][0]) + frame_idx
                         frame.save(frame_format.format(frame_number))
             return output_path