Skip to content

Commit 9b6750b

Browse files
authored
Use cuda 11.6 for auto (#220)
auto: AUTOMATIC1111/stable-diffusion-webui@98947d1 Closes #218 #219
1 parent 5e3f20b commit 9b6750b

File tree

2 files changed

+11
-10
lines changed

2 files changed

+11
-10
lines changed

docker-compose.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ services:
2727
<<: *base_service
2828
profiles: ["auto"]
2929
build: ./services/AUTOMATIC1111
30-
image: sd-auto:15
30+
image: sd-auto:16
3131
environment:
3232
- CLI_ARGS=--allow-code --medvram --xformers
3333

services/AUTOMATIC1111/Dockerfile

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -34,15 +34,15 @@ SHELL ["/bin/bash", "-ceuxo", "pipefail"]
3434

3535
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
3636

37-
RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
37+
RUN pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
3838

3939
RUN apt-get update && apt install fonts-dejavu-core rsync git jq moreutils -y && apt-get clean
4040

4141

4242
RUN <<EOF
4343
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
4444
cd stable-diffusion-webui
45-
git reset --hard d885a4a57b72152745ca76192ef1bdda29e6461d
45+
git reset --hard 98947d173e3f1667eba29c904f681047dea9de90
4646
pip install -r requirements_versions.txt
4747
EOF
4848

@@ -56,28 +56,29 @@ ENV ROOT=/stable-diffusion-webui
5656

5757
COPY --from=download /git/ ${ROOT}
5858
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/data/* ${ROOT}/interrogate
59-
RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeFormer/requirements.txt
59+
RUN pip install -r ${ROOT}/repositories/CodeFormer/requirements.txt
6060

6161

6262
ARG DEEPDANBOORU="0"
6363
RUN [[ "${DEEPDANBOORU:-0}" == "0" ]] && : || pip install tensorflow-cpu==2.10 tensorflow-io==0.27.0 git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru
6464

65+
RUN pip install opencv-python-headless \
66+
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
67+
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
68+
pyngrok
69+
6570
# Note: don't update the sha of previous versions because the install will take forever
6671
# instead, update the repo state in a later step
6772

68-
ARG SHA=ac085628540d0ec6a988fad93f5b8f2154209571
73+
ARG SHA=98947d173e3f1667eba29c904f681047dea9de90
6974
RUN <<EOF
7075
cd stable-diffusion-webui
7176
git fetch
7277
git reset --hard ${SHA}
7378
pip install -r requirements_versions.txt
7479
EOF
7580

76-
RUN pip install opencv-python-headless \
77-
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
78-
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
79-
pyngrok
80-
81+
RUN pip install opencv-python-headless
8182

8283
COPY . /docker
8384

0 commit comments

Comments
 (0)