Update pipeline.py

pipeline.py

@@ -268,30 +268,26 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
                     " the batch size of `prompt`."
                 )
 
-            negative_clip_prompt_embed = self._get_clip_prompt_embeds(
+            negative_pooled_prompt_embeds = self._get_clip_prompt_embeds(
                 prompt=negative_prompt,
                 device=device,
                 num_images_per_prompt=num_images_per_prompt,
             )
             
-            t5_negative_prompt_embed = self._get_t5_prompt_embeds(
+            t5_negative_prompt_embeds = self._get_t5_prompt_embeds(
                 prompt=negative_prompt_2,
                 device=device,
                 num_images_per_prompt=num_images_per_prompt,
                 max_sequence_length=max_sequence_length,
             )
 
-            negative_clip_prompt_embed = torch.nn.functional.pad(
-                negative_clip_prompt_embed,
-                (0, t5_negative_prompt_embed.shape[-1] - negative_clip_prompt_embed.shape[-1]),
+            negative_pooled_prompt_embeds = torch.nn.functional.pad(
+                negative_pooled_prompt_embeds,
+                (0, t5_negative_prompt_embeds.shape[-1] - negative_pooled_prompt_embeds.shape[-1]),
             )
 
             negative_prompt_embeds = torch.cat([negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-2)
 
-            
-            negative_pooled_prompt_embeds = torch.cat(
-                [negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1
-            )
 
         if self.text_encoder is not None:
             if isinstance(self, FluxLoraLoaderMixin) and USE_PEFT_BACKEND:
@@ -309,8 +305,8 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
         pooled_prompt_embeds = pooled_prompt_embeds.repeat(1, num_images_per_prompt, 1)
         pooled_prompt_embeds = pooled_prompt_embeds.view(batch_size * num_images_per_prompt, -1)
 
-        negative_pooled_prompt_embeds = negative_pooled_prompt_embeds.repeat(1, num_images_per_prompt, 1)
-        negative_pooled_prompt_embeds = negative_pooled_prompt_embeds.view(batch_size * num_images_per_prompt, -1)
+        negative_prompt_embeds = negative_pooled_prompt_embeds.repeat(1, num_images_per_prompt, 1)
+        negative_prompt_embeds = negative_pooled_prompt_embeds.view(batch_size * num_images_per_prompt, -1)
 
         return prompt_embeds, pooled_prompt_embeds, text_ids, negative_prompt_embeds, negative_pooled_prompt_embeds