src/transformers/models/jukebox/modeling_jukebox.py/JukeboxAttention/_encoder_len
class JukeboxAttention:    def _encoder_len(self):
        encoder_len = self.encoder_len
        encoder_blocks = (encoder_len // self.blocks) + 1
        return encoder_blocks * self.blocks