/vqgan-jax

JAX implementation of VQGAN

Primary LanguagePython

vqgan-jax

[WIP] JAX implementation of 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 to see how to use this model to encode and reconstruct images.