src/transformers/models/tvlt/modeling_tvlt.py/TvltMAEHead/__init__
class TvltMAEHead:    def __init__(self, config, output_dim=None):
        super().__init__()
        self.config = config
        self.decoder = nn.Linear(config.decoder_hidden_size, output_dim)