pip install -r requirements.txt
streamlit run main.py
The first time it runs, it will download the model from Hugging Face automatically.
Images are automatical.ly saved in and outputs/
folder, along with their prompt.
Out of memory errors occur frequently in my setup (8GB GPU), even though I've tried to limit the usage of cache and clean up the CUDA. They are quickly fixed with a clear cache and rerun in Streamlit, conveniently accessible on the C and R keys.