| # 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" | |
| ``` | |