src/transformers/models/albert/modeling_albert.py/AlbertForPreTraining/get_output_embeddings
class AlbertForPreTraining:    def get_output_embeddings(self) -> nn.Linear:
        return self.predictions.decoder