Răsfoiți Sursa

Update versions (#146)

- auto:
https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/36a0ba357ab0742c3c4a28437b68fb29a235afbe
  - History tab should be working, closes #138 
- hlky:
https://github.com/sd-webui/stable-diffusion-webui/commit/bd57d22f2e0ec7eed084bdf2a4c0dcc0ed745daf
- experimental support for the streamlit UI, use the env var
`USE_STREAMLIT`, see the `docker-compose.yml` file, closes #105
  - Don't create issues if it fails, it is still very early in dev.
AbdBarho 4 ani în urmă
părinte
comite
c1e13867d9
4 a modificat fișierele cu 22 adăugiri și 8 ștergeri
  1. 1 0
      docker-compose.yml
  2. 2 2
      services/AUTOMATIC1111/Dockerfile
  3. 9 6
      services/hlky/Dockerfile
  4. 10 0
      services/hlky/run.sh

+ 1 - 0
docker-compose.yml

@@ -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

+ 2 - 2
services/AUTOMATIC1111/Dockerfile

@@ -53,7 +53,7 @@ RUN apt-get install jq moreutils -y
 # Note: don't update the sha of previous versions because the install will take forever
 # Note: don't update the sha of previous versions because the install will take forever
 # instead, update the repo state in a later step
 # instead, update the repo state in a later step
 
 
-ARG SHA=03d62538aebeff51713619fe808c953bdb70193d
+ARG SHA=36a0ba357ab0742c3c4a28437b68fb29a235afbe
 RUN <<EOF
 RUN <<EOF
 cd stable-diffusion-webui
 cd stable-diffusion-webui
 git pull --rebase
 git pull --rebase
@@ -79,4 +79,4 @@ WORKDIR ${WORKDIR}
 EXPOSE 7860
 EXPOSE 7860
 # run, -u to not buffer stdout / stderr
 # run, -u to not buffer stdout / stderr
 CMD /docker/mount.sh && \
 CMD /docker/mount.sh && \
-  python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config --ckpt-dir ${ROOT}/models/Stable-diffusion ${CLI_ARGS}
+  python3 -u ../../webui.py --listen --port 7860 --ckpt-dir ${ROOT}/models/Stable-diffusion ${CLI_ARGS}

+ 9 - 6
services/hlky/Dockerfile

@@ -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 && \
-  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
+CMD /docker/mount.sh && /docker/run.sh

+ 10 - 0
services/hlky/run.sh

@@ -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