src/transformers/models/vilt/modeling_vilt.py/ViltAttention/__init__
class ViltAttention:    def __init__(self, config):
        super().__init__()
        self.attention = ViltSelfAttention(config)
        self.output = ViltSelfOutput(config)
        self.pruned_heads = set()