src/transformers/models/vilt/modeling_vilt.py/ViltForImageAndTextRetrieval/__init__
class ViltForImageAndTextRetrieval:    def __init__(self, config):
        super().__init__(config)

        self.vilt = ViltModel(config)

        # Classifier head
        self.rank_output = nn.Linear(config.hidden_size, 1)

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