src/transformers/models/levit/modeling_levit.py/LevitAttention/train
class LevitAttention:    def train(self, mode=True):
        super().train(mode)
        if mode and self.attention_bias_cache:
            self.attention_bias_cache = {}  # clear ab cache