File tree Expand file tree Collapse file tree 4 files changed +9
-9
lines changed Expand file tree Collapse file tree 4 files changed +9
-9
lines changed Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ services:
42
42
<< : *base_service
43
43
profiles : ["hlky"]
44
44
build : ./services/hlky/
45
- image : sd-hlky:9
45
+ image : sd-hlky:10
46
46
environment :
47
47
- CLI_ARGS=--optimized-turbo
48
48
- USE_STREAMLIT=0
@@ -51,7 +51,7 @@ services:
51
51
<< : *base_service
52
52
profiles : ["lstein"]
53
53
build : ./services/lstein/
54
- image : sd-lstein:7
54
+ image : sd-lstein:8
55
55
environment :
56
56
- PRELOAD=true
57
57
- CLI_ARGS=--max_loaded_models=1
Original file line number Diff line number Diff line change 25
25
26
26
RUN apt-get update && apt install libsndfile1 ffmpeg -y && apt-get clean
27
27
28
- ARG BRANCH=dev SHA=269107a104fc9fee3201eb2c56cf7adb3d063e4b
28
+ ARG BRANCH=dev SHA=18a3b809275c395b9a2730c78d6bc0f9b06671e1
29
29
RUN <<EOF
30
30
cd stable-diffusion
31
31
git fetch
Original file line number Diff line number Diff line change @@ -23,7 +23,7 @@ pip install -r requirements.txt
23
23
EOF
24
24
25
25
26
- ARG BRANCH=development SHA=2b7e3abe57963d199f1d825ddef87ae154c81045
26
+ ARG BRANCH=development SHA=a9aa4e45aa6f5d5a2aa385349131d8733dd380fa
27
27
RUN <<EOF
28
28
git fetch
29
29
git reset --hard
@@ -32,11 +32,12 @@ git reset --hard ${SHA}
32
32
pip install -r requirements.txt
33
33
EOF
34
34
35
- RUN pip uninstall opencv-python -y && pip install --force-reinstall opencv-python-headless==4.5.5.64
35
+ RUN pip install --force-reinstall opencv-python-headless==4.5.5.64
36
36
37
37
COPY . /docker/
38
38
RUN <<EOF
39
39
python3 /docker/info.py /stable-diffusion/frontend/dist/index.html
40
+ touch ~/.invokeai
40
41
EOF
41
42
42
43
@@ -45,4 +46,4 @@ EXPOSE 7860
45
46
46
47
47
48
ENTRYPOINT ["/docker/entrypoint.sh" ]
48
- CMD python3 -u scripts/invoke.py --outdir /output -- web --host 0.0.0.0 --port 7860 ${CLI_ARGS}
49
+ CMD python3 -u scripts/invoke.py --web --host 0.0.0.0 --port 7860 --config /docker/models.yaml --root_dir . --outdir /output ${CLI_ARGS}
Original file line number Diff line number Diff line change @@ -19,12 +19,11 @@ MOUNTS["${ROOT}/models/bert-base-uncased"]=/data/.cache/huggingface/transformers
19
19
MOUNTS[" ${ROOT} /models/openai/clip-vit-large-patch14" ]=/data/.cache/huggingface/transformers
20
20
MOUNTS[" ${ROOT} /models/CompVis/stable-diffusion-safety-checker" ]=/data/.cache/huggingface/transformers
21
21
22
- MOUNTS[ " ${ROOT} /configs/models.yaml " ]=/docker/models.yaml
22
+
23
23
# hacks
24
24
MOUNTS[" /opt/conda/lib/python3.10/site-packages/facexlib/weights" ]=/data/.cache/
25
25
MOUNTS[" ${ROOT} /models/clipseg" ]=/data/.cache/invoke/clipseg/
26
26
27
- # MOUNTS["/opt/conda/lib/python3.9/site-packages/realesrgan/weights"]=/data/RealESRGAN
28
27
29
28
for to_path in " ${! MOUNTS[@]} " ; do
30
29
set -Eeuo pipefail
@@ -41,7 +40,7 @@ for to_path in "${!MOUNTS[@]}"; do
41
40
done
42
41
43
42
if " ${PRELOAD} " == " true" ; then
44
- python3 -u scripts/preload_models.py --no-interactive
43
+ python3 -u scripts/preload_models.py --no-interactive --root . --config_file /docker/models.yaml
45
44
fi
46
45
47
46
exec " $@ "
You can’t perform that action at this time.
0 commit comments