src/transformers/models/jukebox/modeling_jukebox.py/JukeboxMLP/forward
class JukeboxMLP:    def forward(self, hidden_states):
        hidden_states = self.c_fc(hidden_states)
        hidden_states = self.act(hidden_states)
        hidden_states = self.c_proj(hidden_states)
        hidden_states = self.dropout(hidden_states)
        return hidden_states