patil-suraj / vit-vqgan

JAX implementation ViT-VQGAN

MIT License

77 stars 11 forks source link

readme

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.