src/transformers/models/roformer/modeling_tf_roformer.py/TFRoFormerForCausalLM/get_lm_head
class TFRoFormerForCausalLM:    def get_lm_head(self) -> tf.keras.layers.Layer:
        return self.mlm.predictions