Skip to content

Commit 4b17fa2

Browse files
authored
fix flux type hint (#12089)
fix-flux-type-hint
1 parent d45199a commit 4b17fa2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_flux.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,7 @@ def forward(
384384
temb: torch.Tensor,
385385
image_rotary_emb: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,
386386
joint_attention_kwargs: Optional[Dict[str, Any]] = None,
387-
) -> torch.Tensor:
387+
) -> Tuple[torch.Tensor, torch.Tensor]:
388388
text_seq_len = encoder_hidden_states.shape[1]
389389
hidden_states = torch.cat([encoder_hidden_states, hidden_states], dim=1)
390390

0 commit comments

Comments
 (0)