File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/diffusers/models/transformers Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 ]:
You can’t perform that action at this time.
0 commit comments