src/transformers/models/deit/modeling_deit.py/DeiTAttention/__init__
class DeiTAttention:    def __init__(self, config: DeiTConfig) -> None:
        super().__init__()
        self.attention = DeiTSelfAttention(config)
        self.output = DeiTSelfOutput(config)
        self.pruned_heads = set()