mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2026-02-03 06:04:24 +01:00
39 lines
1.4 KiB
Docker
39 lines
1.4 KiB
Docker
FROM alpine:3.17 as xformers
|
|
RUN apk add --no-cache aria2
|
|
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'
|
|
|
|
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
|
|
|
RUN apt-get update && apt-get install -y git && apt-get clean
|
|
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
--mount=type=bind,from=xformers,source=/wheel.whl,target=/xformers-0.0.21-cp310-cp310-linux_x86_64.whl \
|
|
pip install /xformers-0.0.21-cp310-cp310-linux_x86_64.whl
|
|
|
|
ENV ROOT=/stable-diffusion
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
|
|
cd ${ROOT} && \
|
|
git checkout master && \
|
|
git reset --hard 91ed2815d542c96fdad75edba2205140de3cbba6 && \
|
|
pip install -r requirements.txt
|
|
|
|
WORKDIR ${ROOT}
|
|
|
|
ADD https://github.com/madebyollin/taesd/raw/main/taesd_decoder.pth \
|
|
${ROOT}/models/vae_approx/taesd_decoder.pth
|
|
ADD https://github.com/madebyollin/taesd/raw/main/taesdxl_decoder.pth \
|
|
${ROOT}/models/vae_approx/taesdxl_decoder.pth
|
|
|
|
# add info
|
|
COPY . /docker/
|
|
RUN cp /docker/extra_model_paths.yaml ${ROOT}
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
|
EXPOSE 7860
|
|
ENTRYPOINT ["/docker/entrypoint.sh"]
|
|
CMD python -u main.py --listen --port 7860 ${CLI_ARGS}
|