Skip to content

Commit 5c465e4

Browse files
committed
fix-copies
1 parent b6dcedc commit 5c465e4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1672,7 +1672,6 @@ def forward(
16721672
encoder_hidden_states: Optional[torch.FloatTensor] = None,
16731673
attention_mask: Optional[torch.FloatTensor] = None,
16741674
cross_attention_kwargs: Optional[Dict[str, Any]] = None,
1675-
encoder_attention_mask: Optional[torch.FloatTensor] = None,
16761675
):
16771676
cross_attention_kwargs = cross_attention_kwargs if cross_attention_kwargs is not None else {}
16781677
hidden_states = self.resnets[0](hidden_states, temb)
@@ -1681,7 +1680,7 @@ def forward(
16811680
hidden_states = attn(
16821681
hidden_states,
16831682
encoder_hidden_states=encoder_hidden_states,
1684-
attention_mask=attention_mask if encoder_hidden_states is None else encoder_attention_mask,
1683+
attention_mask=attention_mask,
16851684
**cross_attention_kwargs,
16861685
)
16871686

0 commit comments

Comments
 (0)