src/transformers/models/megatron_bert/modeling_megatron_bert.py/MegatronBertForPreTraining/__init__
class MegatronBertForPreTraining:    def __init__(self, config, add_binary_head=True):
        super().__init__(config)

        self.bert = MegatronBertModel(config)
        self.cls = MegatronBertPreTrainingHeads(config)

        self.init_weights()