5698c49653
- auto:050a6a798c- Now uses python 3.10 - requires a complete re-install - Image is now smaller (5.7GB vs 9.8GB) - hlky:fe6e72fde7- lstein:31869885d9- img2img now works
46 lines
1.4 KiB
Docker
46 lines
1.4 KiB
Docker
# syntax=docker/dockerfile:1
|
|
|
|
FROM continuumio/miniconda3:4.12.0
|
|
|
|
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
RUN conda install python=3.8.5 && conda clean -a -y
|
|
RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
|
|
|
|
RUN apt-get update && apt install fonts-dejavu-core rsync gcc -y && apt-get clean
|
|
|
|
|
|
RUN <<EOF
|
|
git config --global http.postBuffer 1048576000
|
|
git clone https://github.com/sd-webui/stable-diffusion-webui.git stable-diffusion
|
|
cd stable-diffusion
|
|
git reset --hard 1a9c053cb7b6832695771db2555c0adc9b41e95f
|
|
conda env update --file environment.yaml -n base
|
|
conda clean -a -y
|
|
EOF
|
|
|
|
|
|
ARG BRANCH=dev SHA=fe6e72fde79214aa848d47afe74487d9855b8643
|
|
RUN <<EOF
|
|
cd stable-diffusion
|
|
git fetch
|
|
git checkout ${BRANCH}
|
|
git reset --hard ${SHA}
|
|
conda env update --file environment.yaml -n base
|
|
conda clean -a -y
|
|
EOF
|
|
|
|
# add info
|
|
COPY . /docker/
|
|
RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh
|
|
|
|
WORKDIR /stable-diffusion
|
|
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
|
EXPOSE 7860
|
|
|
|
CMD /docker/mount.sh && \
|
|
python3 -u scripts/webui.py --outdir /output --ckpt /data/StableDiffusion/model.ckpt ${CLI_ARGS}
|
|
# sed -i -- 's/8501/7860/g' .streamlit/config.toml && STREAMLIT_SERVER_HEADLESS=true python -u -m streamlit run scripts/webui_streamlit.py --theme.base dark
|