src/transformers/models/mobilevit/modeling_mobilevit.py/MobileViTAttention/__init__
class MobileViTAttention:    def __init__(self, config: MobileViTConfig, hidden_size: int) -> None:
        super().__init__()
        self.attention = MobileViTSelfAttention(config, hidden_size)
        self.output = MobileViTSelfOutput(config, hidden_size)
        self.pruned_heads = set()