Skip to content

Commit 2af7baa

Browse files
authored
Remove *pooled_* mentions from Chroma inpaint (#13026)
Remove `*pooled_*` mentions from Chroma as it has just one TE.
1 parent a7cb14e commit 2af7baa

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

src/diffusers/pipelines/chroma/pipeline_chroma_inpainting.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -482,8 +482,6 @@ def check_inputs(
482482
negative_prompt=None,
483483
prompt_embeds=None,
484484
negative_prompt_embeds=None,
485-
pooled_prompt_embeds=None,
486-
negative_pooled_prompt_embeds=None,
487485
callback_on_step_end_tensor_inputs=None,
488486
padding_mask_crop=None,
489487
max_sequence_length=None,
@@ -531,15 +529,6 @@ def check_inputs(
531529
f" {negative_prompt_embeds.shape}."
532530
)
533531

534-
if prompt_embeds is not None and pooled_prompt_embeds is None:
535-
raise ValueError(
536-
"If `prompt_embeds` are provided, `pooled_prompt_embeds` also have to be passed. Make sure to generate `pooled_prompt_embeds` from the same text encoder that was used to generate `prompt_embeds`."
537-
)
538-
if negative_prompt_embeds is not None and negative_pooled_prompt_embeds is None:
539-
raise ValueError(
540-
"If `negative_prompt_embeds` are provided, `negative_pooled_prompt_embeds` also have to be passed. Make sure to generate `negative_pooled_prompt_embeds` from the same text encoder that was used to generate `negative_prompt_embeds`."
541-
)
542-
543532
if prompt_embeds is not None and prompt_attention_mask is None:
544533
raise ValueError("Cannot provide `prompt_embeds` without also providing `prompt_attention_mask")
545534

@@ -793,13 +782,11 @@ def __call__(
793782
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
794783
latents: Optional[torch.FloatTensor] = None,
795784
prompt_embeds: Optional[torch.FloatTensor] = None,
796-
pooled_prompt_embeds: Optional[torch.FloatTensor] = None,
797785
ip_adapter_image: Optional[PipelineImageInput] = None,
798786
ip_adapter_image_embeds: Optional[List[torch.Tensor]] = None,
799787
negative_ip_adapter_image: Optional[PipelineImageInput] = None,
800788
negative_ip_adapter_image_embeds: Optional[List[torch.Tensor]] = None,
801789
negative_prompt_embeds: Optional[torch.FloatTensor] = None,
802-
negative_pooled_prompt_embeds: Optional[torch.FloatTensor] = None,
803790
output_type: Optional[str] = "pil",
804791
return_dict: bool = True,
805792
joint_attention_kwargs: Optional[Dict[str, Any]] = None,

0 commit comments

Comments
 (0)