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

Added build args and new profile

Browse source
This commit is contained in:
Imaginator 2022-10-17 14:36:04 -04:00
parent ef7e5d7bfb
commit f06947e97e
2 changed files with 17 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
docker-compose.yml
View file

@ -35,6 +35,16 @@ services:
<<: *base_service
profiles: ["auto"]
build: ./services/AUTOMATIC1111
environment:
- CLI_ARGS=--allow-code --medvram --xformers
auto-deepdanbooru:
<<: *automatic
profiles: ["auto-deepdanbooru"]
build:
context: ./services/AUTOMATIC1111
args:
- DEEPDANBOORU=1
environment:
- CLI_ARGS=--allow-code --medvram --xformers --deepdanbooru

10
services/AUTOMATIC1111/Dockerfile
View file

@ -73,10 +73,14 @@ RUN pip install --prefer-binary --no-cache-dir opencv-python-headless \
COPY --from=xformers /wheel.whl xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
RUN pip install xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
ARG DEEPDANBOORU=0
RUN <<EOF
pip install tensorflow
pip install tensorflow-io
pip install git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru deepdanbooru
if ["$DEEPDANBOORU" = "1"]
then
pip install tensorflow
pip install tensorflow-io
pip install git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru deepdanbooru
fi
EOF
COPY . /docker