src/transformers/models/jukebox/modeling_jukebox.py/JukeboxConditionalAutoregressive/split_chunks
class JukeboxConditionalAutoregressive:    def split_chunks(self, length, chunk_size):
        n_passes = (length + chunk_size - 1) // chunk_size
        chunk_sizes = [*[chunk_size] * (n_passes - 1), (length - 1) % chunk_size + 1]
        return chunk_sizes