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