AbdBarho/stable-diffusion-webui-docker
1
0
Fork 0
mirror of https://github.com/AbdBarho/stable-diffusion-webui-docker.git synced 2026-02-04 06:34:19 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

experimental streamlit support

Browse source
This commit is contained in:
AbdBarho 2022-10-16 16:06:02 +02:00
parent 34076ed401
commit 661d1bea79
3 changed files with 20 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
docker-compose.yml
View file

@ -29,6 +29,7 @@ services:
build: ./services/hlky/ build: ./services/hlky/
environment: environment:
- CLI_ARGS=--optimized-turbo - CLI_ARGS=--optimized-turbo
- USE_STREAMLIT=0
auto: &automatic auto: &automatic
<<: *base_service <<: *base_service

15
services/hlky/Dockerfile
View file

@ -22,7 +22,7 @@ conda clean -a -y
EOF EOF
ARG BRANCH=dev SHA=fd51bab1ec03543eb57faa954caab879ef2bafe7 ARG BRANCH=dev SHA=bd57d22f2e0ec7eed084bdf2a4c0dcc0ed745daf
RUN <<EOF RUN <<EOF
cd stable-diffusion cd stable-diffusion
git fetch git fetch
@ -34,12 +34,15 @@ EOF
# add info # add info
COPY . /docker/ COPY . /docker/
RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh RUN <<EOF
python /docker/info.py /stable-diffusion/frontend/frontend.py
chmod +x /docker/mount.sh /docker/run.sh
# streamlit
sed -i -- 's/8501/7860/g' /stable-diffusion/.streamlit/config.toml
EOF
WORKDIR /stable-diffusion WORKDIR /stable-diffusion
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS="" ENV PYTHONPATH="${PYTHONPATH}:${PWD}" STREAMLIT_SERVER_HEADLESS=true USE_STREAMLIT=0 CLI_ARGS=""
EXPOSE 7860 EXPOSE 7860
CMD /docker/mount.sh && \ CMD /docker/mount.sh && /docker/run.sh
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
# 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

10
services/hlky/run.sh Executable file
View file

@ -0,0 +1,10 @@
#!/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
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
fi