stable-diffusion-webui-docker/services/hlky/Dockerfile

46 lines
1.4 KiB
Docker
Raw Normal View History

2022-08-27 11:38:12 +00:00
# syntax=docker/dockerfile:1
FROM continuumio/miniconda3:4.12.0
2022-09-03 15:36:19 +00:00
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
ENV DEBIAN_FRONTEND=noninteractive
2022-08-27 11:38:12 +00:00
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
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-03 15:36:19 +00:00
RUN <<EOF
2022-09-16 04:51:14 +00:00
git config --global http.postBuffer 1048576000
2022-09-07 17:45:03 +00:00
git clone https://github.com/sd-webui/stable-diffusion-webui.git stable-diffusion
2022-09-03 15:36:19 +00:00
cd stable-diffusion
git reset --hard 1a9c053cb7b6832695771db2555c0adc9b41e95f
2022-09-03 15:36:19 +00:00
conda env update --file environment.yaml -n base
conda clean -a -y
EOF
ARG BRANCH=dev SHA=fd51bab1ec03543eb57faa954caab879ef2bafe7
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
2022-09-03 15:36:19 +00:00
2022-08-28 17:09:15 +00:00
# add info
COPY . /docker/
2022-09-07 17:45:03 +00:00
RUN python /docker/info.py /stable-diffusion/frontend/frontend.py && chmod +x /docker/mount.sh
2022-08-28 17:09:15 +00:00
2022-08-27 11:38:12 +00:00
WORKDIR /stable-diffusion
ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
2022-08-27 11:38:12 +00:00
EXPOSE 7860
2022-09-14 17:48:00 +00:00
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