File tree 1 file changed +1
-2
lines changed
src/diffusers/pipelines/versatile_diffusion
1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -1672,7 +1672,6 @@ def forward(
1672
1672
encoder_hidden_states : Optional [torch .FloatTensor ] = None ,
1673
1673
attention_mask : Optional [torch .FloatTensor ] = None ,
1674
1674
cross_attention_kwargs : Optional [Dict [str , Any ]] = None ,
1675
- encoder_attention_mask : Optional [torch .FloatTensor ] = None ,
1676
1675
):
1677
1676
cross_attention_kwargs = cross_attention_kwargs if cross_attention_kwargs is not None else {}
1678
1677
hidden_states = self .resnets [0 ](hidden_states , temb )
@@ -1681,7 +1680,7 @@ def forward(
1681
1680
hidden_states = attn (
1682
1681
hidden_states ,
1683
1682
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 ,
1685
1684
** cross_attention_kwargs ,
1686
1685
)
1687
1686
You can’t perform that action at this time.
0 commit comments