NadaGh's picture
End of training
dde5d93 verified

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"