src/transformers/models/flava/modeling_flava.py/FlavaAttention/__init__
class FlavaAttention:    def __init__(self, config: FlavaPossibleConfigs) -> None:
        super().__init__()
        self.attention = FlavaSelfAttention(config)
        self.output = FlavaSelfOutput(config)
        self.pruned_heads = set()