Sampling New Images from VQ-VAE

I understand how to implement the VQ-VAE in PyTorch but can’t find a well-documented PyTorch implementation of how we can sample from the VQ-VAE model. That is the PixelCNN. Can anyone recommend one to me? Thanks

Thanks