src/transformers/models/nezha/modeling_nezha.py/NezhaOnlyMLMHead/__init__
class NezhaOnlyMLMHead:    def __init__(self, config):
        super().__init__()
        self.predictions = NezhaLMPredictionHead(config)