Update pipeline.py

pipeline.py

@@ -852,8 +852,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
 
         image = image.to(device=device, dtype=dtype)
 
-        if do_classifier_free_guidance and not guess_mode:
-            image = torch.cat([image] * 2)
+        # if do_classifier_free_guidance and not guess_mode:
+        #     image = torch.cat([image] * 2)
             
         print("prepared control image_batch_size", image.shape)
         print("prepared control device", image.device)
@@ -1265,6 +1265,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     if self.controlnet != None:
                         
                         current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :] 
+                        current_context_conditioning_frames = torch.cat([current_context_conditioning_frames] * 2) if do_classifier_free_guidance else current_context_conditioning_frames
+
                     
                         if guess_mode and self.do_classifier_free_guidance:
                             # Infer ControlNet only for the conditional batch.