As mentioned in the paper, VQ-VAE needs to be trained before the diffusion model can be trained. The instructions say that to train VQ-VAE, the main.py file has to be run. @ThomasMrY, can you please look into the issue? It seems like the main.py file is not uploaded by mistake