AbdBarho/stable-diffusion-webui-docker
1
0
Fork 0
mirror of https://github.com/AbdBarho/stable-diffusion-webui-docker.git synced 2026-02-03 22:24:19 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Simplified requirement installing, startup.sh

Browse source
This commit is contained in:
PassiveLemon 2023-05-08 14:04:24 -04:00
parent 17c44cabcb
commit 89092c1d47
2 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
services/comfy/Dockerfile
View file

@ -7,7 +7,7 @@ FROM python:3.10.9-slim
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
RUN apt-get update && apt-get install -y git aria2 && apt-get clean
RUN apt-get update && apt-get install -y git aria2 build-essential && apt-get clean
RUN --mount=type=cache,target=/cache --mount=type=cache,target=/root/.cache/pip \
aria2c -x 5 --dir /cache --out torch-2.0.0-cp310-cp310-linux_x86_64.whl -c \

20
services/comfy/entrypoint.sh
View file

@ -44,16 +44,16 @@ for to_path in "${!MOUNTS[@]}"; do
echo Mounted $(basename "${from_path}")
done
if [ "$(ls -A /stable-diffusion/custom_nodes)" ]; then
chmod 777 -R "/stable-diffusion/custom_nodes/"
apt-get install build-essential -y
for dir in "/stable-diffusion/custom_nodes/*"; do
if [ -e "$dir/requirements.txt" ]; then
echo $dir
cd $dir
pip install -r requirements.txt
fi
done
if [ -f "/data/config/comfy/startup.sh" ]; then
pushd ${ROOT}
. /data/config/comfy/startup.sh
popd
fi
shopt -s nullglob
list=(${ROOT}/custom_nodes/*/requirements.txt)
for req in "${list[@]}"; do
pip install -r "$req"
done
exec "$@"