diff --git a/src/transformers/modeling_bart.py b/src/transformers/modeling_bart.py index a851d649..3e5a27b9 100644 --- a/src/transformers/modeling_bart.py +++ b/src/transformers/modeling_bart.py @@ -687,7 +687,6 @@ class SelfAttention(nn.Module): static_kv: bool, ) -> Optional[Tensor]: # saved key padding masks have shape (bsz, seq_len) - if prev_key_padding_mask is not None and static_kv: new_key_padding_mask = prev_key_padding_mask elif prev_key_padding_mask is not None and key_padding_mask is not None: