Update README.md
Browse files
README.md
CHANGED
|
@@ -5,12 +5,6 @@ license: apache-2.0
|
|
| 5 |
## PSNR
|
| 6 |
36.20
|
| 7 |
|
| 8 |
-
## Usage
|
| 9 |
-
```
|
| 10 |
-
from diffusers import AutoencoderKL
|
| 11 |
-
|
| 12 |
-
vae = AutoencoderKL.from_pretrained("AiArtLab/simplevae",subfolder="vae").cuda().half()
|
| 13 |
-
```
|
| 14 |
|
| 15 |
## === Absolute ===
|
| 16 |
```
|
|
@@ -47,6 +41,14 @@ AiArtLab/simplevae | MSE=4.818e-04 PSNR=36.20 LPIPS=0.035 Edge=0.095 KL=
|
|
| 47 |
|
| 48 |
https://imgsli.com/NDE1MzE0/5/2
|
| 49 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 50 |
## VAE Training Process
|
| 51 |
|
| 52 |
- Inited from AuraDiffusion/16ch-vae (not compatible), added mid block/retrained
|
|
|
|
| 5 |
## PSNR
|
| 6 |
36.20
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
## === Absolute ===
|
| 10 |
```
|
|
|
|
| 41 |
|
| 42 |
https://imgsli.com/NDE1MzE0/5/2
|
| 43 |
|
| 44 |
+
### Diffusers
|
| 45 |
+
```
|
| 46 |
+
from diffusers import AutoencoderKL
|
| 47 |
+
|
| 48 |
+
vae = AutoencoderKL.from_pretrained("AiArtLab/sdxl_vae",subfolder="vae").cuda().half()
|
| 49 |
+
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
## VAE Training Process
|
| 53 |
|
| 54 |
- Inited from AuraDiffusion/16ch-vae (not compatible), added mid block/retrained
|