mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-25 15:55:18 +00:00
88d1879d59
Add a GitHub Action that builds a docker image on every push and tag, and publishes it to GitHub container repository, and -- if a docker username is defined in the repository secrets -- to the Docker Hub.
73 lines
2.0 KiB
Docker
73 lines
2.0 KiB
Docker
# syntax=docker/dockerfile:1
|
|
|
|
ARG PYTHON_VERSION=3.11
|
|
|
|
FROM python:${PYTHON_VERSION}-slim
|
|
|
|
ARG PYTORCH_INSTALL_ARGS=""
|
|
ARG EXTRA_ARGS=""
|
|
ARG USERNAME=comfyui
|
|
ARG USER_UID=1000
|
|
ARG USER_GID=${USER_UID}
|
|
|
|
# Fail fast on errors or unset variables
|
|
SHELL ["/bin/bash", "-eux", "-o", "pipefail", "-c"]
|
|
|
|
RUN <<EOF
|
|
groupadd --gid ${USER_GID} ${USERNAME}
|
|
useradd --uid ${USER_UID} --gid ${USER_GID} -m ${USERNAME}
|
|
EOF
|
|
|
|
RUN <<EOF
|
|
apt-get update
|
|
apt-get install -y --no-install-recommends \
|
|
git \
|
|
git-lfs \
|
|
rsync \
|
|
fonts-recommended
|
|
EOF
|
|
|
|
# run instructions as user
|
|
USER ${USER_UID}:${USER_GID}
|
|
|
|
WORKDIR /app
|
|
|
|
ENV XDG_CACHE_HOME=/cache
|
|
ENV PIP_CACHE_DIR=/cache/pip
|
|
ENV VIRTUAL_ENV=/app/venv
|
|
ENV VIRTUAL_ENV_CUSTOM=/app/custom_venv
|
|
|
|
# create cache directory. During build we will use a cache mount,
|
|
# but later this is useful for custom node installs
|
|
RUN --mount=type=cache,target=/cache/,uid=${USER_UID},gid=${USER_GID} \
|
|
mkdir -p ${PIP_CACHE_DIR}
|
|
|
|
# create virtual environment to manage packages
|
|
RUN python -m venv ${VIRTUAL_ENV}
|
|
|
|
# run python from venv (prefer custom_venv over baked-in one)
|
|
ENV PATH="${VIRTUAL_ENV_CUSTOM}/bin:${VIRTUAL_ENV}/bin:${PATH}"
|
|
|
|
RUN --mount=type=cache,target=/cache/,uid=${USER_UID},gid=${USER_GID} \
|
|
pip install torch torchvision torchaudio ${PYTORCH_INSTALL_ARGS}
|
|
|
|
# copy requirements files first so packages can be cached separately
|
|
COPY --chown=${USER_UID}:${USER_GID} requirements.txt .
|
|
RUN --mount=type=cache,target=/cache/,uid=${USER_UID},gid=${USER_GID} \
|
|
pip install -r requirements.txt
|
|
|
|
COPY --chown=${USER_UID}:${USER_GID} . .
|
|
|
|
# default environment variables
|
|
ENV COMFYUI_ADDRESS=0.0.0.0
|
|
ENV COMFYUI_PORT=8188
|
|
ENV COMFYUI_EXTRA_BUILD_ARGS="${EXTRA_ARGS}"
|
|
ENV COMFYUI_EXTRA_ARGS=""
|
|
# default start command
|
|
CMD \
|
|
if [ -d "${VIRTUAL_ENV_CUSTOM}" ]; then \
|
|
rsync -aP "${VIRTUAL_ENV}/" "${VIRTUAL_ENV_CUSTOM}/" ;\
|
|
sed -i "s!${VIRTUAL_ENV}!${VIRTUAL_ENV_CUSTOM}!g" "${VIRTUAL_ENV_CUSTOM}/pyvenv.cfg" ;\
|
|
fi ;\
|
|
python -u main.py --listen ${COMFYUI_ADDRESS} --port ${COMFYUI_PORT} ${COMFYUI_EXTRA_BUILD_ARGS} ${COMFYUI_EXTRA_ARGS}
|