VQ Vae vs Vae Math