Commit 4b17fa2

authored

fix flux type hint (#12089)

fix-flux-type-hint

1 parent d45199a commit 4b17fa2Copy full SHA for 4b17fa2

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -384,7 +384,7 @@ def forward(`
`384`	`384`	`temb: torch.Tensor,`
`385`	`385`	`image_rotary_emb: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,`
`386`	`386`	`joint_attention_kwargs: Optional[Dict[str, Any]] = None,`
`387`		`- ) -> torch.Tensor:`
	`387`	`+ ) -> Tuple[torch.Tensor, torch.Tensor]:`
`388`	`388`	`text_seq_len = encoder_hidden_states.shape[1]`
`389`	`389`	`hidden_states = torch.cat([encoder_hidden_states, hidden_states], dim=1)`
`390`	`390`

Comments

(0)