src/transformers/models/oneformer/modeling_oneformer.py/OneFormerTaskModel/forward
class OneFormerTaskModel:    def forward(self, inputs: Tensor) -> Tensor:
        task_tokens = self.task_mlp(inputs.float())
        return task_tokens