Skip to content

Commit 22c503f

Browse files
leffffyiyixuxu
andauthored
Update src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py
Co-authored-by: YiYi Xu <[email protected]>
1 parent 54cf03c commit 22c503f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -564,7 +564,11 @@ def __call__(
564564
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
565565
latents: Optional[torch.Tensor] = None,
566566
prompt_embeds_qwen: Optional[torch.Tensor] = None,
567-
negative_prompt_embeds: Optional[torch.Tensor] = None,
567+
prompt_embeds_clip: Optional[torch.Tensor] = None,
568+
negative_prompt_embeds_qwen: Optional[torch.Tensor] = None,
569+
negative_prompt_embeds_clip: Optional[torch.Tensor] = None,
570+
prompt_cu_seqlens: Optional[torch.Tensor] = None,
571+
negative_prompt_cu_seqlens: Optional[torch.Tensor] = None,
568572
output_type: Optional[str] = "pil",
569573
return_dict: bool = True,
570574
callback_on_step_end: Optional[

0 commit comments

Comments
 (0)