src/transformers/models/oneformer/modeling_oneformer.py/PredictionBlock/forward
class PredictionBlock:    def forward(self, input: Tensor) -> Tensor:
        hidden_state = input
        for layer in self.layers:
            hidden_state = layer(hidden_state)
        return hidden_state