src/transformers/models/electra/modeling_electra.py/ElectraForMaskedLM/set_output_embeddings
class ElectraForMaskedLM:    def set_output_embeddings(self, word_embeddings):
        self.generator_lm_head = word_embeddings