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.
59 lines
1.7 KiB
Docker
59 lines
1.7 KiB
Docker
# syntax=docker/dockerfile:1
|
|
|
|
FROM python:3.10-slim
|
|
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1 PIP_NO_CACHE_DIR=1
|
|
|
|
|
|
RUN pip install torch==1.12.0+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
|
|
|
|
RUN apt-get update && apt-get install git -y && apt-get clean
|
|
|
|
RUN git clone https://github.com/invoke-ai/InvokeAI.git /stable-diffusion
|
|
|
|
WORKDIR /stable-diffusion
|
|
|
|
RUN <<EOF
|
|
git reset --hard 5c31feb3a1096d437c94b6e1c3224eb7a7224a85
|
|
git config --global http.postBuffer 1048576000
|
|
pip install -r binary_installer/py3.10-linux-x86_64-cuda-reqs.txt
|
|
EOF
|
|
|
|
|
|
# patch match:
|
|
# https://github.com/invoke-ai/InvokeAI/blob/main/docs/installation/INSTALL_PATCHMATCH.md
|
|
RUN <<EOF
|
|
apt-get update
|
|
# apt-get install build-essential python3-opencv libopencv-dev -y
|
|
apt-get install make g++ libopencv-dev -y
|
|
apt-get clean
|
|
cd /usr/lib/x86_64-linux-gnu/pkgconfig/
|
|
ln -sf opencv4.pc opencv.pc
|
|
EOF
|
|
|
|
ARG BRANCH=main SHA=38cd968130e386d188bdef68f9fbfbbbfabb2da0
|
|
RUN <<EOF
|
|
git fetch
|
|
git reset --hard
|
|
git checkout ${BRANCH}
|
|
git reset --hard ${SHA}
|
|
pip install -r binary_installer/py3.10-linux-x86_64-cuda-reqs.txt
|
|
EOF
|
|
|
|
RUN pip install --force-reinstall opencv-python-headless && python3 -c "from patchmatch import patch_match"
|
|
|
|
|
|
COPY . /docker/
|
|
RUN <<EOF
|
|
python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
|
|
touch ~/.invokeai
|
|
EOF
|
|
|
|
|
|
ENV ROOT=/stable-diffusion PYTHONPATH="${PYTHONPATH}:${ROOT}" PRELOAD=false CLI_ARGS=""
|
|
EXPOSE 7860
|
|
|
|
ENTRYPOINT ["/docker/entrypoint.sh"]
|
|
CMD python3 -u scripts/invoke.py --web --host 0.0.0.0 --port 7860 --config /docker/models.yaml --root_dir ${ROOT} --outdir /output/invoke ${CLI_ARGS}
|