Skip to content

patil-suraj/vqgan-jax

Folders and files

NameName
Last commit message
Last commit date

Latest commit

10ef240 · Jul 9, 2022

History

18 Commits
Jul 9, 2022
Jun 28, 2021
Jun 28, 2021
Jul 17, 2021

Repository files navigation

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.