src/transformers/models/fnet/modeling_fnet.py/FNetOnlyMLMHead/__init__
class FNetOnlyMLMHead:    def __init__(self, config):
        super().__init__()
        self.predictions = FNetLMPredictionHead(config)