src/transformers/models/mobilebert/modeling_mobilebert.py/MobileBertForMaskedLM/__init__
class MobileBertForMaskedLM:    def __init__(self, config):
        super().__init__(config)
        self.mobilebert = MobileBertModel(config, add_pooling_layer=False)
        self.cls = MobileBertOnlyMLMHead(config)
        self.config = config

        self.init_weights()