src/transformers/models/bert/modeling_bert.py/BertForPreTraining/__init__
class BertForPreTraining:    def __init__(self, config):
        super().__init__(config)

        self.bert = BertModel(config)
        self.cls = BertPreTrainingHeads(config)

        self.init_weights()