mirror of
https://github.com/huggingface/diffusers.git
synced 2026-06-16 00:35:12 +08:00
2e2684f014
* Add vae_roundtrip.py example * Add cuda support to vae_roundtrip * Move vae_roundtrip.py into research_projects/vae * Fix channel scaling in vae roundrip and also support taesd. * Apply ruff --fix for CI gatekeep check --------- Co-authored-by: Álvaro Somoza <asomoza@users.noreply.github.com>
12 lines
385 B
Markdown
12 lines
385 B
Markdown
# VAE
|
|
|
|
`vae_roundtrip.py` Demonstrates the use of a VAE by roundtripping an image through the encoder and decoder. Original and reconstructed images are displayed side by side.
|
|
|
|
```
|
|
cd examples/research_projects/vae
|
|
python vae_roundtrip.py \
|
|
--pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5" \
|
|
--subfolder="vae" \
|
|
--input_image="/path/to/your/input.png"
|
|
```
|