vqgan-jax icon indicating copy to clipboard operation
vqgan-jax copied to clipboard

JAX implementation of VQGAN

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.