Update pipeline.py

pipeline.py

@@ -1536,9 +1536,11 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
             
             # if we had more then one prompt, we need to offset the video frames back by number of inference steps
             if len(prompt_embeds_list) > 1:
+                
                 # wrap the first n number of frames to the end of the video to fix the offseting from the context scheduler
                 offset_frames = num_inference_steps
-                video_tensor = torch.cat((latents[:, :, offset_frames:, :, :], latents[:, :, :offset_frames, :, :]), dim=2)
+                print("Offsetting video frames by ", offset_frames)
+                latents = torch.cat((latents[:, :, offset_frames:, :, :], latents[:, :, :offset_frames, :, :]), dim=2)
             
             for batch in range((num_frames + output_batch_size - 1) // output_batch_size):
                 start_id = batch * output_batch_size