src/transformers/models/albert/modeling_albert.py/AlbertForMaskedLM/set_output_embeddings
class AlbertForMaskedLM:    def set_output_embeddings(self, new_embeddings: nn.Linear) -> None:
        self.predictions.decoder = new_embeddings