parent
7a1e52bc7a
commit
8df9d10a58
@ -42,7 +42,7 @@ services:
|
|||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["hlky"]
|
profiles: ["hlky"]
|
||||||
build: ./services/hlky/
|
build: ./services/hlky/
|
||||||
image: sd-hlky:9
|
image: sd-hlky:10
|
||||||
environment:
|
environment:
|
||||||
- CLI_ARGS=--optimized-turbo
|
- CLI_ARGS=--optimized-turbo
|
||||||
- USE_STREAMLIT=0
|
- USE_STREAMLIT=0
|
||||||
@ -51,7 +51,7 @@ services:
|
|||||||
<<: *base_service
|
<<: *base_service
|
||||||
profiles: ["lstein"]
|
profiles: ["lstein"]
|
||||||
build: ./services/lstein/
|
build: ./services/lstein/
|
||||||
image: sd-lstein:7
|
image: sd-lstein:8
|
||||||
environment:
|
environment:
|
||||||
- PRELOAD=true
|
- PRELOAD=true
|
||||||
- CLI_ARGS=--max_loaded_models=1
|
- CLI_ARGS=--max_loaded_models=1
|
||||||
|
@ -25,7 +25,7 @@ EOF
|
|||||||
|
|
||||||
RUN apt-get update && apt install libsndfile1 ffmpeg -y && apt-get clean
|
RUN apt-get update && apt install libsndfile1 ffmpeg -y && apt-get clean
|
||||||
|
|
||||||
ARG BRANCH=dev SHA=269107a104fc9fee3201eb2c56cf7adb3d063e4b
|
ARG BRANCH=dev SHA=18a3b809275c395b9a2730c78d6bc0f9b06671e1
|
||||||
RUN <<EOF
|
RUN <<EOF
|
||||||
cd stable-diffusion
|
cd stable-diffusion
|
||||||
git fetch
|
git fetch
|
||||||
|
@ -23,7 +23,7 @@ pip install -r requirements.txt
|
|||||||
EOF
|
EOF
|
||||||
|
|
||||||
|
|
||||||
ARG BRANCH=development SHA=2b7e3abe57963d199f1d825ddef87ae154c81045
|
ARG BRANCH=development SHA=a9aa4e45aa6f5d5a2aa385349131d8733dd380fa
|
||||||
RUN <<EOF
|
RUN <<EOF
|
||||||
git fetch
|
git fetch
|
||||||
git reset --hard
|
git reset --hard
|
||||||
@ -32,11 +32,12 @@ git reset --hard ${SHA}
|
|||||||
pip install -r requirements.txt
|
pip install -r requirements.txt
|
||||||
EOF
|
EOF
|
||||||
|
|
||||||
RUN pip uninstall opencv-python -y && pip install --force-reinstall opencv-python-headless==4.5.5.64
|
RUN pip install --force-reinstall opencv-python-headless==4.5.5.64
|
||||||
|
|
||||||
COPY . /docker/
|
COPY . /docker/
|
||||||
RUN <<EOF
|
RUN <<EOF
|
||||||
python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
|
python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
|
||||||
|
touch ~/.invokeai
|
||||||
EOF
|
EOF
|
||||||
|
|
||||||
|
|
||||||
@ -45,4 +46,4 @@ EXPOSE 7860
|
|||||||
|
|
||||||
|
|
||||||
ENTRYPOINT ["/docker/entrypoint.sh"]
|
ENTRYPOINT ["/docker/entrypoint.sh"]
|
||||||
CMD python3 -u scripts/invoke.py --outdir /output --web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
|
CMD python3 -u scripts/invoke.py --web --host 0.0.0.0 --port 7860 --config /docker/models.yaml --root_dir . --outdir /output ${CLI_ARGS}
|
||||||
|
@ -19,12 +19,11 @@ MOUNTS["${ROOT}/models/bert-base-uncased"]=/data/.cache/huggingface/transformers
|
|||||||
MOUNTS["${ROOT}/models/openai/clip-vit-large-patch14"]=/data/.cache/huggingface/transformers
|
MOUNTS["${ROOT}/models/openai/clip-vit-large-patch14"]=/data/.cache/huggingface/transformers
|
||||||
MOUNTS["${ROOT}/models/CompVis/stable-diffusion-safety-checker"]=/data/.cache/huggingface/transformers
|
MOUNTS["${ROOT}/models/CompVis/stable-diffusion-safety-checker"]=/data/.cache/huggingface/transformers
|
||||||
|
|
||||||
MOUNTS["${ROOT}/configs/models.yaml"]=/docker/models.yaml
|
|
||||||
# hacks
|
# hacks
|
||||||
MOUNTS["/opt/conda/lib/python3.10/site-packages/facexlib/weights"]=/data/.cache/
|
MOUNTS["/opt/conda/lib/python3.10/site-packages/facexlib/weights"]=/data/.cache/
|
||||||
MOUNTS["${ROOT}/models/clipseg"]=/data/.cache/invoke/clipseg/
|
MOUNTS["${ROOT}/models/clipseg"]=/data/.cache/invoke/clipseg/
|
||||||
|
|
||||||
# MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN
|
|
||||||
|
|
||||||
for to_path in "${!MOUNTS[@]}"; do
|
for to_path in "${!MOUNTS[@]}"; do
|
||||||
set -Eeuo pipefail
|
set -Eeuo pipefail
|
||||||
@ -41,7 +40,7 @@ for to_path in "${!MOUNTS[@]}"; do
|
|||||||
done
|
done
|
||||||
|
|
||||||
if "${PRELOAD}" == "true"; then
|
if "${PRELOAD}" == "true"; then
|
||||||
python3 -u scripts/preload_models.py --no-interactive
|
python3 -u scripts/preload_models.py --no-interactive --root . --config_file /docker/models.yaml
|
||||||
fi
|
fi
|
||||||
|
|
||||||
exec "$@"
|
exec "$@"
|
||||||
|
Loading…
Reference in New Issue
Block a user