tokenizer/vqgan/README.md

Pretrained VQVAE Models

install

pip install omegaconf
pip install einops

• download all needed models from https://github.com/CompVis/taming-transformers and put in pretrained_models/
• pip install pytorch_lightning
• python3 tools/convert_pytorch_lightning_to_torch.py
• pip uninstall pytorch_lightning

demo

cd ${THIS_REPO_ROOT}
python3 tokenizer/vqgan/taming_vqgan_demo.py

acknowledge

Codes in this folder are modified from from https://github.com/CompVis/taming-transformers