smoothieAI commited on
Commit
eb41c97
·
1 Parent(s): af0c644

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -783,7 +783,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
783
  end_id = min((batch + 1) * output_batch_size, num_frames)
784
  video_tensor = self.decode_latents(latents[:, :, start_id:end_id, :, :])
785
  video = tensor2vid(video_tensor, self.image_processor, output_type=output_type)
786
- frame_list = video[0][0]
787
  print("frame_list type", type(frame_list))
788
  for f_id, frame in enumerate(frame_list):
789
  frame_number = start_id + f_id
 
783
  end_id = min((batch + 1) * output_batch_size, num_frames)
784
  video_tensor = self.decode_latents(latents[:, :, start_id:end_id, :, :])
785
  video = tensor2vid(video_tensor, self.image_processor, output_type=output_type)
786
+ frame_list = video[0]
787
  print("frame_list type", type(frame_list))
788
  for f_id, frame in enumerate(frame_list):
789
  frame_number = start_id + f_id