Fix breaking change with diffusers qwen image

2026-06-26 10:04:35 -06:00 · 2026-06-26 10:04:35 -06:00 · 4a99ddabad
parent 5f04ae7ad5
commit 4a99ddabad
1 changed files with 0 additions and 2 deletions
--- a/extensions_built_in/diffusion_models/qwen_image/qwen_image.py
+++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image.py
@ -333,7 +333,6 @@ class QwenImageModel(BaseModel):
        prompt_embeds_mask = text_embeddings.attention_mask.to(
            self.device_torch, dtype=torch.int64
        )
-        txt_seq_lens = prompt_embeds_mask.sum(dim=1).tolist()

        noise_pred = self.transformer(
            hidden_states=latent_model_input.to(
@ -344,7 +343,6 @@ class QwenImageModel(BaseModel):
            encoder_hidden_states=enc_hs.detach(),
            encoder_hidden_states_mask=prompt_embeds_mask.detach(),
            img_shapes=img_shapes,
-            txt_seq_lens=txt_seq_lens,
            return_dict=False,
            **kwargs,
        )[0]