Skip to content

patil-suraj/vit-vqgan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

JAX implementation ViT-VQGAN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published