src/transformers/models/jukebox/modeling_jukebox.py/JukeboxResnet1D/forward
class JukeboxResnet1D:    def forward(self, hidden_states):
        for block in self.resnet_block:
            hidden_states = block(hidden_states)
        return hidden_states