src/transformers/models/roformer/modeling_roformer.py/RoFormerAttention/__init__
class RoFormerAttention:    def __init__(self, config):
        super().__init__()
        self.self = RoFormerSelfAttention(config)
        self.output = RoFormerSelfOutput(config)
        self.pruned_heads = set()