def indices_to_level_indices(self, indices):
        """Converts indices to indices at each level, perhaps needed for a transformer with factorized embeddings"""
        indices = mx.reshape(indices, (indices.shape[0], -1, 1))
        codes_non_centered = (indices // self._basis) % self._levels
        return codes_non_centered
Converts indices to indices at each level, perhaps needed for a transformer with factorized embeddings
mlx_audio/tts/models/spark/modules/finite_scalar_quantization.py