Skip to content

Commit 87bbde4

Browse files
committed
fix type hints
1 parent 2f86879 commit 87bbde4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/models/transformers/transformer_qwenimage.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,7 @@ def __call__(
492492
encoder_hidden_states_mask: torch.FloatTensor = None,
493493
attention_mask: Optional[torch.FloatTensor] = None,
494494
image_rotary_emb: Optional[torch.Tensor] = None,
495-
encoder_hidden_states_len: Optional[torch.Tensor] = None,
495+
encoder_hidden_states_len: Optional[List[int]] = None,
496496
) -> torch.FloatTensor:
497497
if encoder_hidden_states is None:
498498
raise ValueError("QwenDoubleStreamAttnProcessor2_0 requires encoder_hidden_states (text stream)")
@@ -698,7 +698,7 @@ def forward(
698698
encoder_hidden_states_mask: torch.Tensor,
699699
temb: torch.Tensor,
700700
image_rotary_emb: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,
701-
encoder_hidden_states_len: Optional[torch.Tensor] = None,
701+
encoder_hidden_states_len: Optional[List[int]] = None,
702702
joint_attention_kwargs: Optional[Dict[str, Any]] = None,
703703
modulate_index: Optional[List[int]] = None,
704704
) -> Tuple[torch.Tensor, torch.Tensor]:

0 commit comments

Comments
 (0)