src/transformers/models/levit/modeling_levit.py/LevitStage/forward
class LevitStage:    def forward(self, hidden_state):
        for layer in self.layers:
            hidden_state = layer(hidden_state)
        return hidden_state