src/transformers/models/ernie/modeling_ernie.py/ErnieOnlyMLMHead/__init__
class ErnieOnlyMLMHead:    def __init__(self, config):
        super().__init__()
        self.predictions = ErnieLMPredictionHead(config)