src/transformers/models/jukebox/modeling_jukebox.py/JukeboxBottleneckBlock/dequantise
class JukeboxBottleneckBlock:    def dequantise(self, music_tokens):
        dequantised_states = F.embedding(music_tokens, self.codebook)
        return dequantised_states