Skip to content

Commit 66056f1

Browse files
committed
fix error if no attn mask is passed
1 parent 87bbde4 commit 66056f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_qwenimage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -953,7 +953,7 @@ def forward(
953953
encoder_hidden_states = self.txt_in(encoder_hidden_states)
954954

955955
# Use the encoder_hidden_states sequence length for RoPE computation and normalize mask
956-
if torch.all(encoder_hidden_states_mask):
956+
if encoder_hidden_states_mask is not None and torch.all(encoder_hidden_states_mask):
957957
encoder_hidden_states_mask = None
958958
text_seq_len, text_seq_len_per_sample, encoder_hidden_states_mask = compute_text_seq_len_from_mask(
959959
encoder_hidden_states, encoder_hidden_states_mask

0 commit comments

Comments
 (0)