src/transformers/models/mobilebert/modeling_mobilebert.py/MobileBertOnlyMLMHead/__init__
class MobileBertOnlyMLMHead:    def __init__(self, config):
        super().__init__()
        self.predictions = MobileBertLMPredictionHead(config)