make qwen hidden states contiguous to make torchao happy.

2026-02-05 02:15:13 +08:00 · 2026-02-04 18:13:26 +05:30
1 changed files with 2 additions and 2 deletions
--- a/src/diffusers/models/transformers/transformer_qwenimage.py
+++ b/src/diffusers/models/transformers/transformer_qwenimage.py
@@ -561,11 +561,11 @@ class QwenDoubleStreamAttnProcessor2_0:
        img_attn_output = joint_hidden_states[:, seq_txt:, :]  # Image part

        # Apply output projections
-        img_attn_output = attn.to_out[0](img_attn_output)
+        img_attn_output = attn.to_out[0](img_attn_output.contiguous())
        if len(attn.to_out) > 1:
            img_attn_output = attn.to_out[1](img_attn_output)  # dropout

-        txt_attn_output = attn.to_add_out(txt_attn_output)
+        txt_attn_output = attn.to_add_out(txt_attn_output.contiguous())

        return img_attn_output, txt_attn_output