src/transformers/models/roberta/modeling_roberta.py/RobertaAttention/__init__
class RobertaAttention:    def __init__(self, config):
        super().__init__()
        self.self = RobertaSelfAttention(config)
        self.output = RobertaSelfOutput(config)
        self.pruned_heads = set()