From f06947e97e61cd8043c17b699e04e454c56b2081 Mon Sep 17 00:00:00 2001
From: Imaginator <mriegel@gmail.com>
Date: Mon, 17 Oct 2022 14:36:04 -0400
Subject: [PATCH] Added build args and new profile

---
 docker-compose.yml                | 10 ++++++++++
 services/AUTOMATIC1111/Dockerfile | 10 +++++++---
 2 files changed, 17 insertions(+), 3 deletions(-)

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 <<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