diff --git a/docker-compose.yml b/docker-compose.yml index 96ea07b..e99080d 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -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 diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile index 1c8a802..421dcc9 100644 --- a/services/AUTOMATIC1111/Dockerfile +++ b/services/AUTOMATIC1111/Dockerfile @@ -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 <