diff --git a/build/Dockerfile b/build/Dockerfile index bbf4c4d..3b9eaf2 100644 --- a/build/Dockerfile +++ b/build/Dockerfile @@ -30,12 +30,14 @@ RUN git clone https://github.com/hlky/stable-diffusion-webui.git && cd stable-di # cd / && rm -rf stable-diffusion-webui +# add info +COPY info.py /info.py +RUN python /info.py /stable-diffusion/scripts/webui.py + WORKDIR /stable-diffusion ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" EXPOSE 7860 -COPY info.py /info.py CMD \ - 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 index 5a6585f..ca479e0 100644 --- a/build/info.py +++ b/build/info.py @@ -8,5 +8,5 @@ file.write_text( Created using stable-diffusion-webui-docker.

For help and advanced usage guides, -""") +""", 1) ) \ No newline at end of file