Skip to content

Commit 0785dba

Browse files
sayakpaulhlky
andauthored
[Docs] Add negative prompt docs to FluxPipeline (#10531)
* add negative_prompt documentation. * add proper docs for negative prompts * fix-copies * remove comment. * Apply suggestions from code review Co-authored-by: hlky <hlky@hlky.ac> * fix-copies --------- Co-authored-by: hlky <hlky@hlky.ac>
1 parent 5cda8ea commit 0785dba

8 files changed

+32
-15
lines changed

src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,9 +404,9 @@ def encode_prompt(
404404
negative_prompt_2 (`str` or `List[str]`, *optional*):
405405
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
406406
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
407-
negative_prompt_2 (`str` or `List[str]`, *optional*):
407+
negative_prompt_3 (`str` or `List[str]`, *optional*):
408408
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
409-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
409+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
410410
prompt_embeds (`torch.FloatTensor`, *optional*):
411411
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
412412
provided, text embeddings will be generated from `prompt` input argument.

src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet_inpainting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -410,9 +410,9 @@ def encode_prompt(
410410
negative_prompt_2 (`str` or `List[str]`, *optional*):
411411
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
412412
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
413-
negative_prompt_2 (`str` or `List[str]`, *optional*):
413+
negative_prompt_3 (`str` or `List[str]`, *optional*):
414414
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
415-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
415+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
416416
prompt_embeds (`torch.FloatTensor`, *optional*):
417417
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
418418
provided, text embeddings will be generated from `prompt` input argument.

src/diffusers/pipelines/flux/pipeline_flux.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -665,7 +665,16 @@ def __call__(
665665
instead.
666666
prompt_2 (`str` or `List[str]`, *optional*):
667667
The prompt or prompts to be sent to `tokenizer_2` and `text_encoder_2`. If not defined, `prompt` is
668-
will be used instead
668+
will be used instead.
669+
negative_prompt (`str` or `List[str]`, *optional*):
670+
The prompt or prompts not to guide the image generation. If not defined, one has to pass
671+
`negative_prompt_embeds` instead. Ignored when not using guidance (i.e., ignored if `true_cfg_scale` is
672+
not greater than `1`).
673+
negative_prompt_2 (`str` or `List[str]`, *optional*):
674+
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
675+
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
676+
true_cfg_scale (`float`, *optional*, defaults to 1.0):
677+
When > 1.0 and a provided `negative_prompt`, enables true classifier-free guidance.
669678
height (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
670679
The height in pixels of the generated image. This is set to 1024 by default for the best results.
671680
width (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
@@ -709,6 +718,14 @@ def __call__(
709718
Pre-generated image embeddings for IP-Adapter. It should be a list of length same as number of
710719
IP-adapters. Each element should be a tensor of shape `(batch_size, num_images, emb_dim)`. If not
711720
provided, embeddings are computed from the `ip_adapter_image` input argument.
721+
negative_prompt_embeds (`torch.FloatTensor`, *optional*):
722+
Pre-generated negative text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt
723+
weighting. If not provided, negative_prompt_embeds will be generated from `negative_prompt` input
724+
argument.
725+
negative_pooled_prompt_embeds (`torch.FloatTensor`, *optional*):
726+
Pre-generated negative pooled text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt
727+
weighting. If not provided, pooled negative_prompt_embeds will be generated from `negative_prompt`
728+
input argument.
712729
output_type (`str`, *optional*, defaults to `"pil"`):
713730
The output format of the generate image. Choose between
714731
[PIL](https://pillow.readthedocs.io/en/stable/): `PIL.Image.Image` or `np.array`.

src/diffusers/pipelines/pag/pipeline_pag_sd_3.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -375,9 +375,9 @@ def encode_prompt(
375375
negative_prompt_2 (`str` or `List[str]`, *optional*):
376376
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
377377
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
378-
negative_prompt_2 (`str` or `List[str]`, *optional*):
378+
negative_prompt_3 (`str` or `List[str]`, *optional*):
379379
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
380-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
380+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
381381
prompt_embeds (`torch.FloatTensor`, *optional*):
382382
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
383383
provided, text embeddings will be generated from `prompt` input argument.

src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -391,9 +391,9 @@ def encode_prompt(
391391
negative_prompt_2 (`str` or `List[str]`, *optional*):
392392
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
393393
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
394-
negative_prompt_2 (`str` or `List[str]`, *optional*):
394+
negative_prompt_3 (`str` or `List[str]`, *optional*):
395395
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
396-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
396+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
397397
prompt_embeds (`torch.FloatTensor`, *optional*):
398398
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
399399
provided, text embeddings will be generated from `prompt` input argument.

src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -383,9 +383,9 @@ def encode_prompt(
383383
negative_prompt_2 (`str` or `List[str]`, *optional*):
384384
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
385385
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
386-
negative_prompt_2 (`str` or `List[str]`, *optional*):
386+
negative_prompt_3 (`str` or `List[str]`, *optional*):
387387
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
388-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
388+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
389389
prompt_embeds (`torch.FloatTensor`, *optional*):
390390
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
391391
provided, text embeddings will be generated from `prompt` input argument.

src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -400,9 +400,9 @@ def encode_prompt(
400400
negative_prompt_2 (`str` or `List[str]`, *optional*):
401401
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
402402
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
403-
negative_prompt_2 (`str` or `List[str]`, *optional*):
403+
negative_prompt_3 (`str` or `List[str]`, *optional*):
404404
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
405-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
405+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
406406
prompt_embeds (`torch.FloatTensor`, *optional*):
407407
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
408408
provided, text embeddings will be generated from `prompt` input argument.

src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_inpaint.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -406,9 +406,9 @@ def encode_prompt(
406406
negative_prompt_2 (`str` or `List[str]`, *optional*):
407407
The prompt or prompts not to guide the image generation to be sent to `tokenizer_2` and
408408
`text_encoder_2`. If not defined, `negative_prompt` is used in all the text-encoders.
409-
negative_prompt_2 (`str` or `List[str]`, *optional*):
409+
negative_prompt_3 (`str` or `List[str]`, *optional*):
410410
The prompt or prompts not to guide the image generation to be sent to `tokenizer_3` and
411-
`text_encoder_3`. If not defined, `negative_prompt` is used in both text-encoders
411+
`text_encoder_3`. If not defined, `negative_prompt` is used in all the text-encoders.
412412
prompt_embeds (`torch.FloatTensor`, *optional*):
413413
Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
414414
provided, text embeddings will be generated from `prompt` input argument.

0 commit comments

Comments
 (0)