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