src/transformers/models/vit/modeling_vit.py/ViTAttention/__init__
class ViTAttention:    def __init__(self, config):
        super().__init__()
        self.attention = ViTSelfAttention(config)
        self.output = ViTSelfOutput(config)
        self.pruned_heads = set()