vit-vqgan

JAX implementation of ViT-VQGAN.

Acknowledgements

Jiahui Yu for his advice on the implementation based on the work on ViT-VQGAN and Parti.
Phil Wang for suggesting using convolutions after self-attention layers.
Katherine Crowson for suggesting using convolutions in feed-forward layers.

Name		Name	Last commit message	Last commit date
Latest commit History 107 Commits
dataset		dataset
training		training
vit_vqgan		vit_vqgan
.gitignore		.gitignore
LICENSE.md		LICENSE.md
Makefile		Makefile
README.md		README.md
setup.py		setup.py