mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
102 lines
2.4 KiB
Docker
102 lines
2.4 KiB
Docker
# syntax=docker/dockerfile:1
|
|
|
|
# Maintained by Matthias Wild <mauwii@outlook.de>
|
|
|
|
ARG PYTHON_VERSION=3.9
|
|
##################
|
|
## base image ##
|
|
##################
|
|
FROM python:${PYTHON_VERSION}-slim AS python-base
|
|
|
|
# prepare for buildkit cache
|
|
RUN rm -f /etc/apt/apt.conf.d/docker-clean
|
|
|
|
# Install necesarry packages
|
|
RUN \
|
|
--mount=type=cache,target=/var/cache/apt,sharing=locked \
|
|
apt-get update \
|
|
&& apt-get install -y \
|
|
--no-install-recommends \
|
|
libgl1-mesa-glx=20.3.* \
|
|
libglib2.0-0=2.66.* \
|
|
libopencv-dev=4.5.* \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# set working directory and path
|
|
ARG APPDIR=/usr/src
|
|
ARG APPNAME=InvokeAI
|
|
WORKDIR ${APPDIR}
|
|
ENV PATH=${APPDIR}/${APPNAME}/bin:$PATH
|
|
|
|
######################
|
|
## build frontend ##
|
|
######################
|
|
FROM node:lts as frontend-builder
|
|
|
|
# Copy Sources
|
|
ARG APPDIR=/usr/src
|
|
ARG FRONTEND_DIR=invokeai/frontend
|
|
WORKDIR ${APPDIR}/${FRONTEND_DIR}
|
|
COPY ${FRONTEND_DIR} .
|
|
|
|
# install dependencies
|
|
RUN \
|
|
--mount=type=cache,target=/usr/local/share/.cache/yarn \
|
|
yarn install \
|
|
--frozen-lockfile \
|
|
--non-interactive
|
|
|
|
# build frontend
|
|
RUN yarn build
|
|
|
|
###################################
|
|
## install python dependencies ##
|
|
###################################
|
|
FROM python-base AS pyproject-builder
|
|
ENV PIP_USE_PEP517=1
|
|
|
|
# Install dependencies
|
|
RUN \
|
|
--mount=type=cache,target=/var/cache/apt,sharing=locked \
|
|
apt-get update \
|
|
&& apt-get install -y \
|
|
--no-install-recommends \
|
|
build-essential=12.9 \
|
|
gcc=4:10.2.* \
|
|
python3-dev=3.9.* \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# create virtual environment
|
|
RUN python3 -m venv "${APPNAME}" \
|
|
--upgrade-deps
|
|
|
|
# copy sources
|
|
COPY --link . .
|
|
ARG FRONTEND_DIR=invokeai/frontend
|
|
COPY --from=frontend-builder ${APPDIR}/${FRONTEND_DIR}/dist ${FRONTEND_DIR}/dist
|
|
|
|
# install pyproject.toml
|
|
ARG PIP_EXTRA_INDEX_URL
|
|
ENV PIP_EXTRA_INDEX_URL ${PIP_EXTRA_INDEX_URL}
|
|
ARG PIP_PACKAGE=.
|
|
RUN --mount=type=cache,target=/root/.cache/pip,sharing=locked \
|
|
"${APPDIR}/${APPNAME}/bin/pip" install ${PIP_PACKAGE}
|
|
|
|
# build patchmatch
|
|
RUN python3 -c "from patchmatch import patch_match"
|
|
|
|
#####################
|
|
## runtime image ##
|
|
#####################
|
|
FROM python-base AS runtime
|
|
|
|
# setup environment
|
|
COPY --from=pyproject-builder ${APPDIR}/${APPNAME} ${APPDIR}/${APPNAME}
|
|
ENV INVOKEAI_ROOT=/data
|
|
ENV INVOKE_MODEL_RECONFIGURE="--yes --default_only"
|
|
|
|
# set Entrypoint and default CMD
|
|
ENTRYPOINT [ "invokeai" ]
|
|
CMD [ "--web", "--host=0.0.0.0" ]
|
|
VOLUME [ "/data" ]
|