2022-09-05 17:51:22 +00:00
|
|
|
# syntax=docker/dockerfile:1
|
|
|
|
|
|
|
|
FROM continuumio/miniconda3:4.12.0
|
|
|
|
|
|
|
|
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
|
|
|
|
2022-09-11 18:18:50 +00:00
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
|
2022-09-18 11:49:06 +00:00
|
|
|
# now it requires python3.9
|
2022-09-19 20:02:46 +00:00
|
|
|
RUN conda install python=3.9 && conda clean -a -y
|
2022-09-05 17:51:22 +00:00
|
|
|
RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
|
|
|
|
|
2022-09-15 18:45:21 +00:00
|
|
|
RUN apt-get update && apt install fonts-dejavu-core rsync gcc -y && apt-get clean
|
2022-09-05 17:51:22 +00:00
|
|
|
|
2022-10-07 07:46:07 +00:00
|
|
|
ENV PIP_EXISTS_ACTION=w
|
2022-09-05 17:51:22 +00:00
|
|
|
|
|
|
|
RUN <<EOF
|
2022-09-21 17:10:27 +00:00
|
|
|
git clone https://github.com/invoke-ai/InvokeAI.git stable-diffusion
|
2022-09-05 17:51:22 +00:00
|
|
|
cd stable-diffusion
|
2022-10-07 07:46:07 +00:00
|
|
|
git reset --hard 8a8be92eac17e0ef699528157596b2336bdee532
|
2022-09-19 20:02:46 +00:00
|
|
|
sed -i -- 's/python=3.8.5/python=3.9/g' environment.yaml
|
2022-10-07 07:46:07 +00:00
|
|
|
git config --global http.postBuffer 1048576000
|
2022-09-19 20:02:46 +00:00
|
|
|
conda env update --file environment.yaml -n base
|
|
|
|
conda clean -a -y
|
2022-09-05 17:51:22 +00:00
|
|
|
EOF
|
|
|
|
|
|
|
|
|
2022-10-09 09:39:31 +00:00
|
|
|
ARG BRANCH=development SHA=31869885d9d3ee1e62c1109fb141e828e9a0e833
|
2022-09-16 14:40:20 +00:00
|
|
|
RUN <<EOF
|
|
|
|
cd stable-diffusion
|
|
|
|
git fetch
|
2022-09-19 20:02:46 +00:00
|
|
|
git reset --hard
|
2022-09-16 14:40:20 +00:00
|
|
|
git checkout ${BRANCH}
|
|
|
|
git reset --hard ${SHA}
|
2022-09-30 07:37:27 +00:00
|
|
|
conda env update --file environment.yml -n base
|
2022-09-16 14:40:20 +00:00
|
|
|
conda clean -a -y
|
|
|
|
EOF
|
|
|
|
|
2022-09-23 14:24:14 +00:00
|
|
|
RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless transformers==4.19.2
|
2022-09-16 14:40:20 +00:00
|
|
|
|
|
|
|
COPY . /docker/
|
2022-09-19 20:02:46 +00:00
|
|
|
RUN <<EOF
|
2022-10-07 07:46:07 +00:00
|
|
|
python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
|
2022-09-19 20:02:46 +00:00
|
|
|
chmod +x /docker/mount.sh
|
|
|
|
EOF
|
|
|
|
|
2022-09-05 17:51:22 +00:00
|
|
|
|
2022-10-07 07:46:07 +00:00
|
|
|
ENV PRELOAD=false CLI_ARGS=""
|
2022-09-16 14:40:20 +00:00
|
|
|
WORKDIR /stable-diffusion
|
2022-09-05 17:51:22 +00:00
|
|
|
EXPOSE 7860
|
2022-09-16 14:40:20 +00:00
|
|
|
|
2022-09-19 20:02:46 +00:00
|
|
|
CMD /docker/mount.sh && \
|
2022-10-07 07:46:07 +00:00
|
|
|
python3 -u scripts/dream.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|