Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

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

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.