src/transformers/models/jukebox/modeling_jukebox.py/JukeboxModel/encode
class JukeboxModel:    def encode(self, input_audio, start_level=0, end_level=None, bs_chunks=1):
        return self.vqvae.encode(input_audio, start_level, end_level, bs_chunks)