| | --- |
| | license: cc |
| | library_name: diffusers |
| | tags: |
| | - art |
| | model-index: |
| | - name: 16ch-VAE |
| | results: |
| | - task: |
| | type: encoder-loss |
| | dataset: |
| | name: yerevann/coco-karpathy |
| | type: image |
| | metrics: |
| | - name: PSNR |
| | type: PSNR |
| | value: 31.5151 |
| | --- |
| | |
| | ## 16ch-VAE |
| |
|
| | > Disclaimer: this VAE is not intended to be a replacement for SD3's VAE since the latent spaces are entirely different. |
| |
|
| | A fully open source 16ch VAE reproduction for the [SD3](https://arxiv.org/abs/2403.03206). Useful for people who are building their own image generation models and need an off-the-shelf VAE. Natively trained in fp16. |
| |
|
| |
|
| | | VAE | rFID | PSNR | LPIPS | |
| | |------------------|--------|---------|--------| |
| | | SD1.5 VAE | 0.3131 | 26.4332 | 0.0328 | |
| | | SDXL VAE | 0.3511 | 26.7577 | 0.032 | |
| | | SD3 VAE | 0.0257 | 30.3231 | 0.0132 | |
| | | [16ch-VAE](https://huggingface.co/AuraDiffusion/16ch-vae) | 0.0667 | 31.5151 | 0.0136 | |
| | | [16ch-VAE with FFT](https://huggingface.co/AuraDiffusion/16ch-vae)* | 0.1584 | 31.0542 | 0.0281 | |
| |
|
| |
|
| | ### Usage |
| |
|
| | Awaiting https://github.com/huggingface/diffusers/pull/8769 in diffusers! |
