File tree Expand file tree Collapse file tree 1 file changed +8
-4
lines changed
src/diffusers/pipelines/flux Expand file tree Collapse file tree 1 file changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -334,6 +334,10 @@ def encode_prompt(
334334 def __call__ (
335335 self ,
336336 image : PipelineImageInput ,
337+ prompt : Union [str , List [str ]] = None ,
338+ prompt_2 : Optional [Union [str , List [str ]]] = None ,
339+ prompt_embeds : Optional [torch .FloatTensor ] = None ,
340+ pooled_prompt_embeds : Optional [torch .FloatTensor ] = None ,
337341 return_dict : bool = True ,
338342 ):
339343 r"""
@@ -378,10 +382,10 @@ def __call__(
378382 pooled_prompt_embeds ,
379383 _ ,
380384 ) = self .encode_prompt (
381- prompt = [ "" ] * batch_size ,
382- prompt_2 = None ,
383- prompt_embeds = None ,
384- pooled_prompt_embeds = None ,
385+ prompt = prompt * batch_size ,
386+ prompt_2 = prompt_2 ,
387+ prompt_embeds = prompt_embeds ,
388+ pooled_prompt_embeds = pooled_prompt_embeds ,
385389 device = device ,
386390 num_images_per_prompt = 1 ,
387391 max_sequence_length = 512 ,
You can’t perform that action at this time.
0 commit comments