Update pipeline.py

pipeline.py

@@ -773,7 +773,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
 
         # save frames
         if output_path is not None:
-            output_batch_size = 10  # prevents out of memory errors with large videos
+            output_batch_size = 2# prevents out of memory errors with large videos
             num_frames = latents.size(2)  # latents' shape is [batch, channels, frames, height, width]
             num_digits = output_path.count('#')  # count the number of '#' characters
             frame_format = output_path.replace('#' * num_digits, '{:0' + str(num_digits) + 'd}')
@@ -785,6 +785,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 video = tensor2vid(video_tensor, self.image_processor, output_type=output_type)
                 frame_list = video[0]
                 for f_id, frame in enumerate(frame_list):
+                    # print frame and batch id
+                    print(f"Frame: {f_id + start_id} Batch: {batch}")
                     frame_number = start_id + f_id
                     frame.save(frame_format.format(frame_number))
             return output_path