Update custom_pipeline.py

custom_pipeline.py

@@ -91,6 +91,7 @@ class FluxWithCFGPipeline(FluxPipeline):
         self.check_inputs(
             prompt,
             prompt_2,
+            negative_prompt,
             height,
             width,
             prompt_embeds=prompt_embeds,
@@ -118,11 +119,7 @@ class FluxWithCFGPipeline(FluxPipeline):
             max_sequence_length=max_sequence_length,
             lora_scale=lora_scale,
         )
-        (
-            negative_prompt_embeds,
-            negative_pooled_prompt_embeds,
-            negative_text_ids,
-        ) = self.encode_prompt(
+        negative_prompt_embeds, negative_pooled_prompt_embeds, negative_text_ids = self.encode_prompt(
             prompt=negative_prompt,
             prompt_2=negative_prompt_2,
             prompt_embeds=negative_prompt_embeds,
@@ -141,6 +138,7 @@ class FluxWithCFGPipeline(FluxPipeline):
             height,
             width,
             prompt_embeds.dtype,
+            negative_prompt_embeds.dtype,
             device,
             generator,
             latents,