src/transformers/models/electra/modeling_electra.py/ElectraForMaskedLM/__init__
class ElectraForMaskedLM:    def __init__(self, config):
        super().__init__(config)

        self.electra = ElectraModel(config)
        self.generator_predictions = ElectraGeneratorPredictions(config)

        self.generator_lm_head = nn.Linear(config.embedding_size, config.vocab_size)
        self.init_weights()