github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed new breaking change in diffusers with with qwen image

Browse Source
This commit is contained in:
Jaret Burkett
2026-03-23 14:10:55 -06:00
parent 5642b656b9
commit 295094b4b5
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
extensions_built_in/diffusion_models/qwen_image/qwen_image.py
View File

@@ -354,12 +354,13 @@ class QwenImageModel(BaseModel):
def get_prompt_embeds(self, prompt: str) -> PromptEmbeds: def get_prompt_embeds(self, prompt: str) -> PromptEmbeds:
if self.pipeline.text_encoder.device != self.device_torch: if self.pipeline.text_encoder.device != self.device_torch:
self.pipeline.text_encoder.to(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 = PromptEmbeds(prompt_embeds)
pe.attention_mask = prompt_embeds_mask pe.attention_mask = prompt_embeds_mask
return pe return pe