https://github.com/patil-suraj/vqgan-jax

JAX implementation of VQGAN
https://github.com/patil-suraj/vqgan-jax

Last synced: about 2 months ago
JSON representation

JAX implementation of VQGAN

• Host: GitHub
• URL: https://github.com/patil-suraj/vqgan-jax
• Owner: patil-suraj
• Created: 2021-06-28T08:51:24.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2022-07-09T21:18:35.000Z (over 2 years ago)
• Last Synced: 2024-10-23T10:00:45.047Z (2 months ago)
• Language: Python
• Homepage:
• Size: 13.7 KB
• Stars: 89
• Watchers: 5
• Forks: 28
• Open Issues: 7
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# vqgan-jax

[WIP] JAX implementation of [taming-transformers](https://github.com/CompVis/taming-transformers) VQGAN.

>(Note: For now this only consits of the `VQModel` for inference. The discriminatior, transformer model and training scripts will be added later.)

Use this [![Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1mdXXsMbV6K_LTvCh3IImRsFIWcKU5m1w?usp=sharing)

 to see how to use this model to encode and reconstruct images.