src/transformers/models/tvlt/modeling_tvlt.py/TvltAttention/__init__
class TvltAttention:    def __init__(self, config):
        super().__init__()
        self.attention = TvltSelfAttention(config)
        self.output = TvltSelfOutput(config)
        self.pruned_heads = set()