Downloads · 30 days
0
eq-vae-ema-ldm
zelaki/eq-vae-ema-ldm
eq-vae-ema-ldm is a machine learning model from zelaki. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Arxiv: https://arxiv.org/abs/2502.09509 <br
Access
Public
Updated Feb 25, 2025
Repo size
1.4 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.ckpt335 MB · 100%
At a glance
- Access
- Public
- Created
- Feb 25, 2025
- Updated
- Feb 25, 2025
- SHA
- 8b75a443
Model card
From the Hugging Face model README
EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling
Arxiv: https://arxiv.org/abs/2502.09509 <br>
EQ-VAE regularizes the latent space of pretrained autoencoders by enforcing equivariance under scaling and rotation transformations.
Model Description
This model is a regularized version of SD-VAE. We finetune it with EQ-VAE regularization for 44 epochs on Imagenet with EMA weights.
Model Usage
These weights are intended to be used with the EQ-VAE codebase or the CompVis Stable Diffusion codebase. If you are looking for the model to use with the 🧨 diffusers library, come here.
Metrics
Reconstruction performance of eq-vae-ema on Imagenet Validation Set.
| Metric | Score |
|---|---|
| FID | 0.552 |
| PSNR | 26.158 |
| LPIPS | 0.133 |
| SSIM | 0.725 |
- Created
- Feb 25, 2025
- Updated
- Feb 25, 2025