Added prompt to Flux Prior Redux.

Author: Skquark

examples/community/pipeline_hunyuandit_differential_img2img.py

@@ -1107,13 +1107,14 @@ def __call__(
                         callback_kwargs[k] = locals()[k]
                     callback_outputs = callback_on_step_end(self, i, t, callback_kwargs)
 
-                    latents = callback_outputs.pop("latents", latents)
-                    prompt_embeds = callback_outputs.pop("prompt_embeds", prompt_embeds)
-                    negative_prompt_embeds = callback_outputs.pop("negative_prompt_embeds", negative_prompt_embeds)
-                    prompt_embeds_2 = callback_outputs.pop("prompt_embeds_2", prompt_embeds_2)
-                    negative_prompt_embeds_2 = callback_outputs.pop(
-                        "negative_prompt_embeds_2", negative_prompt_embeds_2
-                    )
+                    if callback_outputs is not None:
+                        latents = callback_outputs.pop("latents", latents)
+                        prompt_embeds = callback_outputs.pop("prompt_embeds", prompt_embeds)
+                        negative_prompt_embeds = callback_outputs.pop("negative_prompt_embeds", negative_prompt_embeds)
+                        prompt_embeds_2 = callback_outputs.pop("prompt_embeds_2", prompt_embeds_2)
+                        negative_prompt_embeds_2 = callback_outputs.pop(
+                            "negative_prompt_embeds_2", negative_prompt_embeds_2
+                        )
 
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
                     progress_bar.update()

src/diffusers/pipelines/flux/pipeline_flux_prior_redux.py

@@ -334,6 +334,7 @@ def encode_prompt(
     def __call__(
         self,
         image: PipelineImageInput,
+        prompt: Optional[str] = "",
         return_dict: bool = True,
     ):
         r"""
@@ -378,7 +379,7 @@ def __call__(
                 pooled_prompt_embeds,
                 _,
             ) = self.encode_prompt(
-                prompt=[""] * batch_size,
+                prompt=[prompt] * batch_size,
                 prompt_2=None,
                 prompt_embeds=None,
                 pooled_prompt_embeds=None,