Skip to content

Commit 7ac3eff

Browse files
committed
add missing attn.norm_cross check
1 parent d97a32a commit 7ac3eff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/attention_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ def __call__(self, attn: Attention, hidden_states, encoder_hidden_states=None, a
416416

417417
if encoder_hidden_states is None:
418418
encoder_hidden_states = hidden_states
419-
else:
419+
elif attn.norm_cross:
420420
encoder_hidden_states = attn.norm_encoder_hidden_states(encoder_hidden_states)
421421

422422
key = attn.to_k(encoder_hidden_states) + scale * self.to_k_lora(encoder_hidden_states)

0 commit comments

Comments
 (0)