Add pipeline tag and license

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +6 -4
README.md CHANGED
@@ -1,18 +1,21 @@
1
  ---
2
  library_name: diffusers
 
 
3
  ---
 
4
  ## EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling
5
 
 
 
6
  **EQ-VAE** regularizes the latent space of pretrained autoencoders by enforcing equivariance under scaling and rotation transformations.
7
 
8
  ---
9
  #### Model Description
10
  This model is a regularized version of [SD-VAE](https://github.com/CompVis/latent-diffusion). We finetune it with EQ-VAE regularization for 44 epochs on Imagenet with EMA weights.
11
 
12
-
13
  ## Model Usage
14
 
15
-
16
  2. **Loading the Model**
17
  You can load the model from the Hugging Face Hub:
18
  ```python
@@ -28,5 +31,4 @@ Reconstruction performance of eq-vae-ema on Imagenet Validation Set.
28
  | **PSNR** | 26.158 |
29
  | **LPIPS** | 0.133 |
30
  | **SSIM** | 0.725 |
31
- ---
32
-
 
1
  ---
2
  library_name: diffusers
3
+ license: mit
4
+ pipeline_tag: image-to-image
5
  ---
6
+
7
  ## EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling
8
 
9
+ Project page: https://eq-vae.github.io/.
10
+
11
  **EQ-VAE** regularizes the latent space of pretrained autoencoders by enforcing equivariance under scaling and rotation transformations.
12
 
13
  ---
14
  #### Model Description
15
  This model is a regularized version of [SD-VAE](https://github.com/CompVis/latent-diffusion). We finetune it with EQ-VAE regularization for 44 epochs on Imagenet with EMA weights.
16
 
 
17
  ## Model Usage
18
 
 
19
  2. **Loading the Model**
20
  You can load the model from the Hugging Face Hub:
21
  ```python
 
31
  | **PSNR** | 26.158 |
32
  | **LPIPS** | 0.133 |
33
  | **SSIM** | 0.725 |
34
+ ---