2022-10-16 14:27:20 +00:00
|
|
|
#!/bin/bash
|
|
|
|
|
|
|
|
|
|
set -Eeuo pipefail
|
|
|
|
|
|
|
|
|
|
echo "USE_STREAMLIT = ${USE_STREAMLIT}"
|
|
|
|
|
if [ "${USE_STREAMLIT}" == "1" ]; then
|
|
|
|
|
python -u -m streamlit run scripts/webui_streamlit.py
|
|
|
|
|
else
|
2022-10-22 08:44:39 +00:00
|
|
|
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/v1-5-pruned-emaonly.ckpt ${CLI_ARGS}
|
2022-10-16 14:27:20 +00:00
|
|
|
fi
|
