src/transformers/models/beit/modeling_beit.py/BeitAttention/__init__
class BeitAttention:    def __init__(self, config, window_size=None):
        super().__init__()
        self.attention = BeitSelfAttention(config, window_size=window_size)
        self.output = BeitSelfOutput(config)
        self.pruned_heads = set()