src/transformers/models/jukebox/modeling_jukebox.py/JukeboxBottleneck/encode
class JukeboxBottleneck:    def encode(self, raw_audio):
        music_tokens = [
            level_block.encode(hidden_states) for (level_block, hidden_states) in zip(self.level_blocks, raw_audio)
        ]
        return music_tokens