From ef7e5d7bfbbf6e577846fe70cd7b107115e573dc Mon Sep 17 00:00:00 2001
From: Imaginator <mriegel@gmail.com>
Date: Mon, 17 Oct 2022 13:15:25 -0400
Subject: [PATCH] Add support and enable deepdanbooru setting

---
 docker-compose.yml                | 2 +-
 services/AUTOMATIC1111/Dockerfile | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index ec5c4b1..96ea07b 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -36,7 +36,7 @@ services:
     profiles: ["auto"]
     build: ./services/AUTOMATIC1111
     environment:
-      - CLI_ARGS=--allow-code --medvram --xformers
+      - CLI_ARGS=--allow-code --medvram --xformers --deepdanbooru
 
   auto-cpu:
     <<: *automatic
diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile
index c36a90e..1c8a802 100644
--- a/services/AUTOMATIC1111/Dockerfile
+++ b/services/AUTOMATIC1111/Dockerfile
@@ -73,6 +73,12 @@ 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
 
+RUN <<EOF
+pip install tensorflow
+pip install tensorflow-io
+pip install git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru deepdanbooru
+EOF
+
 COPY . /docker
 RUN <<EOF
 chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py