src/transformers/models/albert/modeling_tf_albert.py/TFAlbertForPreTraining/get_lm_head
class TFAlbertForPreTraining:    def get_lm_head(self) -> tf.keras.layers.Layer:
        return self.predictions