src/transformers/models/data2vec/modeling_data2vec_text.py/Data2VecTextAttention/__init__
class Data2VecTextAttention:    def __init__(self, config, position_embedding_type=None):
        super().__init__()
        self.self = Data2VecTextSelfAttention(config, position_embedding_type=position_embedding_type)
        self.output = Data2VecTextSelfOutput(config)
        self.pruned_heads = set()