# Jax/Flax VQ autoencoder for Stable Diffusion

**URL:** <https://discuss.huggingface.co/t/jax-flax-vq-autoencoder-for-stable-diffusion/24887>\
**Category:** 🧨 Diffusers\
**Created:** [October 24, 2022, 1:59pm UTC](https://discuss.huggingface.co/t/jax-flax-vq-autoencoder-for-stable-diffusion/24887 "2022-10-24T13:59:40Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![Egiob](https://avatars.discourse-cdn.com/v4/letter/e/73ab20/32.png) [@Egiob](https://discuss.huggingface.co/u/Egiob)\
**Post date:** [October 24, 2022, 1:59pm UTC](https://discuss.huggingface.co/t/jax-flax-vq-autoencoder-for-stable-diffusion/24887/1 "2022-10-24T13:59:40Z")

</div>

Hey, it seems to me that for Stable-Diffusion, only the encoder with KL penalty have been translated to JAX/Flax, any plan to add support to the one with VQ-regularization? Thanks 🙂