src/transformers/models/vit_mae/modeling_vit_mae.py/ViTMAEForPreTraining/__init__
class ViTMAEForPreTraining:    def __init__(self, config):
        super().__init__(config)
        self.config = config

        self.vit = ViTMAEModel(config)
        self.decoder = ViTMAEDecoder(config, num_patches=self.vit.embeddings.num_patches)

        # Initialize weights and apply final processing
        self.post_init()