smoothieAI commited on
Commit
37b4d60
·
1 Parent(s): ba50290

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -784,7 +784,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
784
  video = tensor2vid(video_tensor, self.image_processor, output_type=output_type)
785
 
786
  for batch_idx, frame_batch in enumerate(video):
787
- for frame_idx, frame in enumerate(frame_batch[0][0]):
788
  frame_number = start_idx + batch_idx * len(frame_batch[0][0]) + frame_idx
789
  frame.save(frame_format.format(frame_number))
790
  return output_path
 
784
  video = tensor2vid(video_tensor, self.image_processor, output_type=output_type)
785
 
786
  for batch_idx, frame_batch in enumerate(video):
787
+ for frame_idx, frame in enumerate(frame_batch[0]):
788
  frame_number = start_idx + batch_idx * len(frame_batch[0][0]) + frame_idx
789
  frame.save(frame_format.format(frame_number))
790
  return output_path