diff --git a/README.md b/README.md index b2323fa..40e221b 100644 Binary files a/README.md and b/README.md differ diff --git a/build/Dockerfile b/build/Dockerfile index 7fb4366..bbf4c4d 100644 --- a/build/Dockerfile +++ b/build/Dockerfile @@ -13,12 +13,12 @@ RUN apt-get update && apt install fonts-dejavu-core && apt-get clean # Note: don't update the sha of previous versions because the install will take forever # instead, update the repo state in a later step -RUN cd stable-diffusion && git pull && git reset --hard 6b442c05296dd350fcf7f752e17c4b4bb015a03b && \ +RUN cd stable-diffusion && git pull && git reset --hard ac31d943ce0c99b7dd4df7e6cb1c14f2e275ca96 && \ conda env update --file environment.yaml --name base && conda clean -a -y -# download latest UI version, update the sha below in case you want some other version +# download dev UI version, update the sha below in case you want some other version RUN git clone https://github.com/hlky/stable-diffusion-webui.git && cd stable-diffusion-webui && \ - git reset --hard 33890563d237889aca7cbc4f04ae42b885aee07e && \ + git reset --hard 244f945c40b8dec3fd1faf8824b99abeb3bc4b97 && \ cp -t /stable-diffusion/scripts/ txt2img.yaml webui.py webui.yaml webui_playground.py && \ cd / && rm -rf stable-diffusion-webui @@ -32,10 +32,10 @@ RUN git clone https://github.com/hlky/stable-diffusion-webui.git && cd stable-di WORKDIR /stable-diffusion ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" - EXPOSE 7860 +COPY info.py /info.py CMD \ - ln -sf /models/model.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt && \ + python /info.py scripts/webui.py && \ ln -sf /models/GFPGANv1.3.pth /stable-diffusion/src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth && \ ln -sf /models/RealESRGAN_x4plus.pth /stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth && \ # force facexlib cache diff --git a/build/info.py b/build/info.py new file mode 100644 index 0000000..5a6585f --- /dev/null +++ b/build/info.py @@ -0,0 +1,12 @@ +import sys +from pathlib import Path + +file = Path(sys.argv[1]) +file.write_text( + file.read_text().replace('

For help and advanced usage guides,', """ +

+ Created using stable-diffusion-webui-docker. +

+

For help and advanced usage guides, +""") +) \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index 18745bd..82304a7 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -11,7 +11,7 @@ services: - ./output:/output - ./models:/models environment: - - CLI_ARGS=--extra-models-cpu + - CLI_ARGS=--extra-models-cpu --optimized-turbo deploy: resources: reservations: