Skip to content

Commit 6e83133

Browse files
leffffyiyixuxu
andauthored
Update src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py
Co-authored-by: YiYi Xu <[email protected]>
1 parent 70cfb9e commit 6e83133

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/kandinsky5/pipeline_kandinsky.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -743,7 +743,7 @@ def __call__(
743743
# Predict noise residual
744744
pred_velocity = self.transformer(
745745
hidden_states=latents.to(dtype),
746-
encoder_hidden_states=prompt_embeds_dict["text_embeds"].to(dtype),
746+
encoder_hidden_states=prompt_embeds_qwen.to(dtype),
747747
pooled_projections=prompt_embeds_dict["pooled_embed"].to(dtype),
748748
timestep=timestep.to(dtype),
749749
visual_rope_pos=visual_rope_pos,

0 commit comments

Comments
 (0)