src/transformers/models/vit_msn/modeling_vit_msn.py/ViTMSNAttention/__init__
class ViTMSNAttention:    def __init__(self, config: ViTMSNConfig) -> None:
        super().__init__()
        self.attention = ViTMSNSelfAttention(config)
        self.output = ViTMSNSelfOutput(config)
        self.pruned_heads = set()