src/transformers/models/squeezebert/modeling_squeezebert.py/SqueezeBertOnlyMLMHead/__init__
class SqueezeBertOnlyMLMHead:    def __init__(self, config):
        super().__init__()
        self.predictions = SqueezeBertLMPredictionHead(config)