src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertOnlyMLMHead/__init__
class MegatronBertOnlyMLMHead:    def __init__(self, config):
        super().__init__()
        self.predictions = MegatronBertLMPredictionHead(config)