You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.9 KiB
Docker
54 lines
1.9 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_EXISTS_ACTION=w PIP_PREFER_BINARY=1
|
|
|
|
# patch match:
|
|
# https://github.com/invoke-ai/InvokeAI/blob/main/docs/installation/INSTALL_PATCHMATCH.md
|
|
RUN --mount=type=cache,target=/var/cache/apt \
|
|
apt-get update && \
|
|
apt-get install make g++ git libopencv-dev -y && \
|
|
apt-get clean && \
|
|
cd /usr/lib/x86_64-linux-gnu/pkgconfig/ && \
|
|
ln -sf opencv4.pc opencv.pc
|
|
|
|
|
|
ENV ROOT=/InvokeAI
|
|
RUN git clone https://github.com/invoke-ai/InvokeAI.git ${ROOT}
|
|
WORKDIR ${ROOT}
|
|
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
git reset --hard f3b2e02921927d9317255b1c3811f47bd40a2bf9 && \
|
|
pip install -e .
|
|
|
|
|
|
ARG BRANCH=main SHA=f3b2e02921927d9317255b1c3811f47bd40a2bf9
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
git fetch && \
|
|
git reset --hard && \
|
|
git checkout ${BRANCH} && \
|
|
git reset --hard ${SHA} && \
|
|
pip install -U -e .
|
|
|
|
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 -U opencv-python-headless triton /xformers-0.0.21-cp310-cp310-linux_x86_64.whl && \
|
|
python3 -c "from patchmatch import patch_match"
|
|
|
|
|
|
COPY . /docker/
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR=/data/config/invoke CLI_ARGS=""
|
|
EXPOSE 7860
|
|
|
|
ENTRYPOINT ["/docker/entrypoint.sh"]
|
|
CMD invokeai --web --host 0.0.0.0 --port 7860 --root_dir ${ROOT} --config ${CONFIG_DIR}/models.yaml \
|
|
--outdir /output/invoke --embedding_directory /data/embeddings/ --lora_directory /data/models/Lora \
|
|
--no-nsfw_checker --no-safety_checker ${CLI_ARGS}
|
|
|