update Dockerfile

- create models.yaml from models.yaml.example
- run preload_models.py with --no-interactive
This commit is contained in:
mauwii 2022-11-01 01:34:20 +01:00 committed by Lincoln Stein
parent 743f605773
commit 2c30555b84

View File

@ -39,12 +39,12 @@ RUN apt-get update \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
# clone repository and create symlinks # clone repository, create models.yaml and create symlinks
ARG invokeai_git=https://github.com/invoke-ai/InvokeAI.git ARG invokeai_git=https://github.com/invoke-ai/InvokeAI.git
ARG project_name=invokeai ARG project_name=invokeai
RUN git clone ${invokeai_git} /${project_name} \ RUN git clone ${invokeai_git} /${project_name} \
&& mkdir /${project_name}/models/ldm/stable-diffusion-v1 \ && cp /${project_name}/configs/models.yaml.example /${project_name}/configs/models.yaml \
&& ln -s /data/models/sd-v1-4.ckpt /${project_name}/models/ldm/stable-diffusion-v1/model.ckpt \ && ln -s /data/models/v1-5-pruned-emaonly.ckpt /${project_name}/models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt \
&& ln -s /data/outputs/ /${project_name}/outputs && ln -s /data/outputs/ /${project_name}/outputs
# set workdir # set workdir
@ -65,7 +65,8 @@ RUN source ${conda_prefix}/etc/profile.d/conda.sh \
&& echo "conda activate ${project_name}" >> ~/.bashrc \ && echo "conda activate ${project_name}" >> ~/.bashrc \
&& ln -s /data/models/GFPGANv1.4.pth ./src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth \ && ln -s /data/models/GFPGANv1.4.pth ./src/gfpgan/experiments/pretrained_models/GFPGANv1.4.pth \
&& conda activate ${project_name} \ && conda activate ${project_name} \
&& python scripts/preload_models.py && python scripts/preload_models.py \
--no-interactive
# Copy entrypoint and set env # Copy entrypoint and set env
ENV CONDA_PREFIX=${conda_prefix} ENV CONDA_PREFIX=${conda_prefix}