src/transformers/models/deit/modeling_tf_deit.py/TFDeiTForMaskedImageModeling/__init__
class TFDeiTForMaskedImageModeling:    def __init__(self, config: DeiTConfig) -> None:
        super().__init__(config)

        self.deit = TFDeiTMainLayer(config, add_pooling_layer=False, use_mask_token=True, name="deit")
        self.decoder = TFDeitDecoder(config, name="decoder")