src/transformers/models/xlm_roberta/modeling_xlm_roberta.py/XLMRobertaAttention/__init__
class XLMRobertaAttention:    def __init__(self, config, position_embedding_type=None):
        super().__init__()
        self.self = XLMRobertaSelfAttention(config, position_embedding_type=position_embedding_type)
        self.output = XLMRobertaSelfOutput(config)
        self.pruned_heads = set()