File tree Expand file tree Collapse file tree 4 files changed +22
-8
lines changed
Expand file tree Collapse file tree 4 files changed +22
-8
lines changed Original file line number Diff line number Diff line change @@ -29,6 +29,7 @@ services:
2929 build : ./services/hlky/
3030 environment :
3131 - CLI_ARGS=--optimized-turbo
32+ - USE_STREAMLIT=0
3233
3334 auto : &automatic
3435 << : *base_service
Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ RUN apt-get install jq moreutils -y
5353# Note: don't update the sha of previous versions because the install will take forever
5454# instead, update the repo state in a later step
5555
56- ARG SHA=03d62538aebeff51713619fe808c953bdb70193d
56+ ARG SHA=36a0ba357ab0742c3c4a28437b68fb29a235afbe
5757RUN <<EOF
5858cd stable-diffusion-webui
5959git pull --rebase
@@ -79,4 +79,4 @@ WORKDIR ${WORKDIR}
7979EXPOSE 7860
8080# run, -u to not buffer stdout / stderr
8181CMD /docker/mount.sh && \
82- python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config -- ckpt-dir ${ROOT}/models/Stable-diffusion ${CLI_ARGS}
82+ python3 -u ../../webui.py --listen --port 7860 --ckpt-dir ${ROOT}/models/Stable-diffusion ${CLI_ARGS}
Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ conda clean -a -y
2222EOF
2323
2424
25- ARG BRANCH=dev SHA=fd51bab1ec03543eb57faa954caab879ef2bafe7
25+ ARG BRANCH=dev SHA=bd57d22f2e0ec7eed084bdf2a4c0dcc0ed745daf
2626RUN <<EOF
2727cd stable-diffusion
2828git fetch
3434
3535# add info
3636COPY . /docker/
37- RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh
37+ RUN <<EOF
38+ python /docker/info.py /stable-diffusion/frontend/frontend.py
39+ chmod +x /docker/mount.sh /docker/run.sh
40+ # streamlit
41+ sed -i -- 's/8501/7860/g' /stable-diffusion/.streamlit/config.toml
42+ EOF
3843
3944WORKDIR /stable-diffusion
40- ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
45+ ENV PYTHONPATH="${PYTHONPATH}:${PWD}" STREAMLIT_SERVER_HEADLESS=true USE_STREAMLIT=0 CLI_ARGS=""
4146EXPOSE 7860
4247
43- CMD /docker/mount.sh && \
44- python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
45- # sed -i -- 's/8501/7860/g' .streamlit/config.toml && STREAMLIT_SERVER_HEADLESS=true python -u -m streamlit run scripts/webui_streamlit.py --theme.base dark
48+ CMD /docker/mount.sh && /docker/run.sh
Original file line number Diff line number Diff line change 1+ #! /bin/bash
2+
3+ set -Eeuo pipefail
4+
5+ echo " USE_STREAMLIT = ${USE_STREAMLIT} "
6+ if [ " ${USE_STREAMLIT} " == " 1" ]; then
7+ python -u -m streamlit run scripts/webui_streamlit.py
8+ else
9+ python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
10+ fi
You can’t perform that action at this time.
0 commit comments