We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9fe0a9c commit a3dcd98Copy full SHA for a3dcd98
src/diffusers/models/transformers/transformer_qwenimage.py
@@ -561,11 +561,11 @@ def __call__(
561
img_attn_output = joint_hidden_states[:, seq_txt:, :] # Image part
562
563
# Apply output projections
564
- img_attn_output = attn.to_out[0](img_attn_output)
+ img_attn_output = attn.to_out[0](img_attn_output.contiguous())
565
if len(attn.to_out) > 1:
566
img_attn_output = attn.to_out[1](img_attn_output) # dropout
567
568
- txt_attn_output = attn.to_add_out(txt_attn_output)
+ txt_attn_output = attn.to_add_out(txt_attn_output.contiguous())
569
570
return img_attn_output, txt_attn_output
571
0 commit comments