Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 493 Bytes

README.md

File metadata and controls

9 lines (6 loc) · 493 Bytes
Raw

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.