discrete VQ-VAE representations