diff --git a/src/diffusers/models/transformers/transformer_flux.py b/src/diffusers/models/transformers/transformer_flux.py index 3a7202d0f43f..608ea7f6e5f0 100644 --- a/src/diffusers/models/transformers/transformer_flux.py +++ b/src/diffusers/models/transformers/transformer_flux.py @@ -490,6 +490,7 @@ def forward( encoder_hidden_states, temb, image_rotary_emb, + joint_attention_kwargs, ) else: @@ -521,6 +522,7 @@ def forward( encoder_hidden_states, temb, image_rotary_emb, + joint_attention_kwargs, ) else: