src/transformers/models/oneformer/modeling_oneformer.py/OneFormerTaskModel/__init__
class OneFormerTaskModel:    def __init__(self, config: OneFormerConfig):
        super().__init__()
        self.task_mlp = OneFormerMLPPredictionHead(
            config.task_seq_len,
            config.hidden_dim,
            config.hidden_dim,
            2,
        )