Skip to content

Commit 211d3dd

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

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -664,13 +664,13 @@ def __call__(
664664
batch_size = prompt_embeds[0].shape[0] if isinstance(prompt_embeds, (list, tuple)) else prompt_embeds.shape[0]
665665

666666
# 3. Encode input prompt
667-
prompt_embeds_dict, prompt_cu_seqlens = self.encode_prompt(
668-
prompt=prompt,
669-
num_videos_per_prompt=num_videos_per_prompt,
670-
max_sequence_length=max_sequence_length,
671-
device=device,
672-
dtype=dtype,
673-
)
667+
if prompt_embeds_qwen is None:
668+
prompt_embeds_qwen, prompt_embeds_clip, prompt_cu_seqlens = self.encode_prompt(
669+
prompt=prompt,
670+
max_sequence_length=max_sequence_length,
671+
device=device,
672+
dtype=dtype,
673+
)
674674

675675

676676
if self.do_classifier_free_guidance:

0 commit comments

Comments
 (0)