Distillation of a tractable model from the VQ-VAE