Skip to content

Commit f6059b7

Browse files
committed
Fix bug in Flux2TransformerBlock (double stream block)
1 parent 6820d6c commit f6059b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_flux2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -433,7 +433,7 @@ def forward(
433433

434434
# Conditioning txt stream
435435
norm_encoder_hidden_states = self.norm1_context(encoder_hidden_states)
436-
norm_encoder_hidden_states = (1 + c_scale_msa) * encoder_hidden_states + c_shift_msa
436+
norm_encoder_hidden_states = (1 + c_scale_msa) * norm_encoder_hidden_states + c_shift_msa
437437

438438
# Attention on concatenated img + txt stream
439439
attention_outputs = self.attn(

0 commit comments

Comments
 (0)