Skip to content

Commit c5d67eb

Browse files
fix: try newer cudnn and stable-diffusion for torch 2.7
1 parent 7fe335c commit c5d67eb

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

Dockerfile

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,7 @@ RUN . /clone.sh clip-interrogator https://github.com/pharmapsychotic/clip-interr
1313
RUN . /clone.sh generative-models https://github.com/Stability-AI/generative-models 45c443b316737a4ab6e40413d7794a7f5657c19f
1414
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets 6f7db241d2f8ba7457bac5ca9753331f0c266917
1515

16-
17-
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
16+
FROM pytorch/pytorch:2.7.1-cuda12.6-cudnn9-runtime
1817

1918
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
2019

@@ -30,7 +29,7 @@ WORKDIR /
3029
RUN --mount=type=cache,target=/root/.cache/pip \
3130
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && \
3231
cd stable-diffusion-webui && \
33-
git reset --hard v1.9.4 && \
32+
git reset --hard dev && \
3433
mkdir -p openai && \
3534
cd openai && \
3635
git clone https://huggingface.co/openai/clip-vit-large-patch14 && \

0 commit comments

Comments
 (0)