Fix encode_prompt bug for no CFG

nom · web-flow · commit 8c8629960ea2 · 2024-09-24T17:25:35.000-04:00
diff --git a/pipeline_flux_controlnet_inpaint.py b/pipeline_flux_controlnet_inpaint.py
@@ -429,6 +429,9 @@ def encode_prompt(
                 max_sequence_length=max_sequence_length,
                 device=device,
             )
+        else:
+            negative_pooled_prompt_embeds = None
+            negative_prompt_embeds = None            
 
         if self.text_encoder is not None:
             if isinstance(self, FluxLoraLoaderMixin) and USE_PEFT_BACKEND:

Original file line number	Diff line number	Diff line change
`@@ -429,6 +429,9 @@ def encode_prompt(`
`429`	`429`	`max_sequence_length=max_sequence_length,`
`430`	`430`	`device=device,`
`431`	`431`	`)`
	`432`	`+ else:`
	`433`	`+ negative_pooled_prompt_embeds = None`
	`434`	`+ negative_prompt_embeds = None`
`432`	`435`
`433`	`436`	`if self.text_encoder is not None:`
`434`	`437`	`if isinstance(self, FluxLoraLoaderMixin) and USE_PEFT_BACKEND:`