src/transformers/models/albert/modeling_albert.py/AlbertModel/set_input_embeddings
class AlbertModel:    def set_input_embeddings(self, value: nn.Embedding) -> None:
        self.embeddings.word_embeddings = value