Commit
·
37b4d60
1
Parent(s):
ba50290
Update pipeline.py
Browse files- 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]
|
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
|