src/transformers/models/rembert/modeling_rembert.py/RemBertAttention/__init__
class RemBertAttention:    def __init__(self, config):
        super().__init__()
        self.self = RemBertSelfAttention(config)
        self.output = RemBertSelfOutput(config)
        self.pruned_heads = set()