## vqgan-jax

Files copied from [patil-suraj/vqgan-jax](https://github.com/patil-suraj/vqgan-jax/tree/main/vqgan_jax)

Required for VQGAN Jax model.