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'


# TODO: next version is going to support pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
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


WORKDIR /
ENV ROOT=/InvokeAI
RUN git clone --depth 1 https://github.com/invoke-ai/InvokeAI.git ${ROOT}

WORKDIR ${ROOT}
RUN --mount=type=cache,target=/root/.cache/pip \
  # dont fetch entire repo, it is HUGE!
  git fetch origin dedead672f2e9f995b6b96d7ada37456dab15b55 --depth=1 && \
  git reset --hard dedead672f2e9f995b6b96d7ada37456dab15b55 && \
  pip install -e .

RUN --mount=type=cache,target=/root/.cache/pip \
  pip install -U opencv-python-headless xformers==0.0.22 && \
  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}