src/transformers/models/vit_mae/modeling_vit_mae.py/ViTMAEAttention/__init__
class ViTMAEAttention:    def __init__(self, config: ViTMAEConfig) -> None:
        super().__init__()
        self.attention = ViTMAESelfAttention(config)
        self.output = ViTMAESelfOutput(config)
        self.pruned_heads = set()