src/transformers/models/roformer/modeling_tf_roformer.py/TFRoFormerMLMHead/__init__
class TFRoFormerMLMHead:    def __init__(self, config: RoFormerConfig, input_embeddings: tf.keras.layers.Layer, **kwargs):
        super().__init__(**kwargs)

        self.predictions = TFRoFormerLMPredictionHead(config, input_embeddings, name="predictions")