From 295094b4b539771ecbf3dcc7fa0fabba16015463 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 23 Mar 2026 14:10:55 -0600 Subject: [PATCH] Fixed new breaking change in diffusers with with qwen image --- .../diffusion_models/qwen_image/qwen_image.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py index 6e6813ff..99ecd6ad 100644 --- a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py +++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py @@ -354,12 +354,13 @@ class QwenImageModel(BaseModel): def get_prompt_embeds(self, prompt: str) -> PromptEmbeds: if self.pipeline.text_encoder.device != self.device_torch: self.pipeline.text_encoder.to(self.device_torch) + + max_sequence_length = 1024 + + prompt_embeds, prompt_embeds_mask = self.pipeline._get_qwen_prompt_embeds(prompt, self.device_torch) + prompt_embeds = prompt_embeds[:, :max_sequence_length] + prompt_embeds_mask = prompt_embeds_mask[:, :max_sequence_length] - prompt_embeds, prompt_embeds_mask = self.pipeline.encode_prompt( - prompt, - device=self.device_torch, - num_images_per_prompt=1, - ) pe = PromptEmbeds(prompt_embeds) pe.attention_mask = prompt_embeds_mask return pe