src/transformers/models/fnet/modeling_fnet.py/FNetForPreTraining/__init__
class FNetForPreTraining:    def __init__(self, config):
        super().__init__(config)

        self.fnet = FNetModel(config)
        self.cls = FNetPreTrainingHeads(config)

        self.init_weights()