Contribution: This model was contribution by williamberman in VQ-diffusion.
TODO: Make nice model readme
