Update Dockerfile

This commit is contained in:
simonmcnair 2025-03-24 21:22:13 +00:00 committed by GitHub
parent 53153c6f32
commit 8fe07a18e3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -17,8 +17,8 @@ RUN . /clone.sh generative-models https://github.com/Stability-AI/generative-mod
RUN . /clone.sh stable-diffusion-webui-assets https://github.com/AUTOMATIC1111/stable-diffusion-webui-assets 6f7db241d2f8ba7457bac5ca9753331f0c266917
#FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
#FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
@ -74,7 +74,7 @@ RUN --mount=type=cache,target=/root/.cache/sd \
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git ${ROOT} && \
cd stable-diffusion-webui && \
git reset --hard v1.10.1 # && \
#pip install -r requirements_versions.txt
pip install -r requirements_versions.txt
RUN pip install --upgrade typing-extensions