src/transformers/models/mpnet/modeling_tf_mpnet.py/TFMPNetForMaskedLM/__init__
class TFMPNetForMaskedLM:    def __init__(self, config, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)

        self.mpnet = TFMPNetMainLayer(config, name="mpnet")
        self.lm_head = TFMPNetLMHead(config, self.mpnet.embeddings, name="lm_head")