f5e9997a55
Core: c0c2a7c0d55561cfb6f42a3681346b9b70749ff1 UI: 49e6178fd82ca736f9bbc621c6b12487c300e493
59 lines
2.8 KiB
Docker
59 lines
2.8 KiB
Docker
# syntax=docker/dockerfile:1
|
|
|
|
FROM continuumio/miniconda3:4.12.0
|
|
|
|
|
|
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 git clone https://github.com/hlky/stable-diffusion.git && cd stable-diffusion && git reset --hard 554bd068e6f2f6bc55449a67fe017ddd77090f28
|
|
RUN conda env update --file stable-diffusion/environment.yaml --name base && conda clean -a -y
|
|
|
|
# fonts for generating the grid
|
|
RUN apt-get update && apt install fonts-dejavu-core && apt-get clean
|
|
|
|
# Note: don't update the sha of previous versions because the install will take forever
|
|
# instead, update the repo state in a later step
|
|
RUN cd stable-diffusion && git pull && git reset --hard c0c2a7c0d55561cfb6f42a3681346b9b70749ff1 && \
|
|
conda env update --file environment.yaml --name base && conda clean -a -y
|
|
|
|
# download dev UI version, update the sha below in case you want some other version
|
|
RUN <<EOF
|
|
git clone https://github.com/hlky/stable-diffusion-webui.git
|
|
cd stable-diffusion-webui
|
|
# map to this file: https://github.com/hlky/stable-diffusion-webui/blob/master/.github/sync.yml
|
|
git reset --hard 49e6178fd82ca736f9bbc621c6b12487c300e493
|
|
cp -t /stable-diffusion/scripts/ webui.py relauncher.py txt2img.yaml
|
|
cp -t /stable-diffusion/configs/webui webui.yaml
|
|
cp -t /stable-diffusion/frontend/ frontend/*
|
|
cd / && rm -rf stable-diffusion-webui
|
|
EOF
|
|
|
|
# For testing different versions:
|
|
# RUN git clone https://github.com/AbdBarho/stable-diffusion-webui.git && cd stable-diffusion-webui && \
|
|
# git checkout respect-cli-params &&\
|
|
# cp -t /stable-diffusion/scripts/ txt2img.yaml webui.py webui.yaml webui_playground.py && \
|
|
# cd / && rm -rf stable-diffusion-webui
|
|
|
|
|
|
# add info
|
|
COPY info.py /info.py
|
|
RUN python /info.py /stable-diffusion/frontend/frontend.py
|
|
|
|
WORKDIR /stable-diffusion
|
|
ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS="" \
|
|
GFPGAN_PATH=/stable-diffusion/src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth \
|
|
RealESRGAN_PATH=/stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth \
|
|
RealESRGAN_ANIME_PATH=/stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus_anime_6B.pth
|
|
EXPOSE 7860
|
|
CMD \
|
|
for path in "${GFPGAN_PATH}" "${RealESRGAN_PATH}" "${RealESRGAN_ANIME_PATH}"; do \
|
|
name=$(basename "${path}"); \
|
|
base=$(dirname "${path}"); \
|
|
test -f "/models/${name}" && mkdir -p "${base}" && ln -sf "/models/${name}" "${path}" && echo "Mounted ${name}";\
|
|
done;\
|
|
# force facexlib cache
|
|
mkdir -p /cache/weights/ && rm -rf /opt/conda/lib/python3.8/site-packages/facexlib/weights && \
|
|
ln -sf /cache/weights/ /opt/conda/lib/python3.8/site-packages/facexlib/ && \
|
|
# run, -u to not buffer stdout / stderr
|
|
python3 -u scripts/webui.py --outdir /output --ckpt /models/model.ckpt --save-metadata ${CLI_ARGS}
|