src/transformers/models/jukebox/modeling_jukebox.py/JukeboxResConv1DBlock/forward
class JukeboxResConv1DBlock:    def forward(self, hidden_states):
        residuals = hidden_states
        hidden_states = self.activation(hidden_states)
        hidden_states = self.conv1d_1(hidden_states)
        hidden_states = self.activation(hidden_states)
        hidden_states = self.conv1d_2(hidden_states)
        return residuals + self.res_scale * hidden_states