Skip to content

Commit f1bf3b0

Browse files
authored
Bump pytorch containers (#697)
Closes #696 Closes #694
1 parent 35a18b3 commit f1bf3b0

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

docker-compose.yml

Lines changed: 2 additions & 2 deletions
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:76
30+
image: sd-auto:77
3131
environment:
3232
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
3333

@@ -42,7 +42,7 @@ services:
4242
<<: *base_service
4343
profiles: ["invoke"]
4444
build: ./services/invoke/
45-
image: sd-invoke:30
45+
image: sd-invoke:31
4646
environment:
4747
- PRELOAD=true
4848
- CLI_ARGS=--xformers

services/AUTOMATIC1111/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ RUN . /clone.sh generative-models https://github.com/Stability-AI/generative-mod
1414
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets 6f7db241d2f8ba7457bac5ca9753331f0c266917
1515

1616

17-
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
17+
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
1818

1919
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
2020

@@ -40,7 +40,7 @@ COPY --from=download /repositories/ ${ROOT}/repositories/
4040
RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/clip_interrogator/data/* ${ROOT}/interrogate
4141

4242
RUN --mount=type=cache,target=/root/.cache/pip \
43-
pip install pyngrok xformers==0.0.23.post1 \
43+
pip install pyngrok xformers==0.0.26.post1 \
4444
git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \
4545
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
4646
git+https://github.com/mlfoundations/open_clip.git@v2.20.0

services/invoke/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ RUN apk add --no-cache aria2
33
RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/6.0.0/xformers-0.0.21.dev544-cp310-cp310-manylinux2014_x86_64-pytorch201.whl'
44

55

6-
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
6+
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
77

88
ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1
99

0 commit comments

Comments
 (0)