mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Thomas Eding 2e2684f014 Add vae_roundtrip.py example (#7104 )

* 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>

2024-07-04 01:53:09 -04:00

README.md

Add vae_roundtrip.py example (#7104 )

2024-07-04 01:53:09 -04:00

vae_roundtrip.py

Add vae_roundtrip.py example (#7104 )

2024-07-04 01:53:09 -04:00

README.md

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"