Deepdanbooru Support (#194)

Builds on top of #150 

Thanks to @pirahtays

Co-authored-by: Imaginator <[email protected]>
Co-authored-by: pirahtays <[email protected]>

Author: 3 people

data/.gitignore

@@ -13,6 +13,7 @@
 /SwinIR
 /ScuNET
 /LDSR
+/Deepdanbooru
 /Hypernetworks
 /VAE
 /embeddings

docker-compose.yml

@@ -27,7 +27,7 @@ services:
     <<: *base_service
     profiles: ["auto"]
     build: ./services/AUTOMATIC1111
-    image: sd-auto:9
+    image: sd-auto:10
     environment:
       - CLI_ARGS=--allow-code --medvram --xformers
 

services/AUTOMATIC1111/Dockerfile

@@ -59,6 +59,9 @@ RUN mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/data/
 RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeFormer/requirements.txt
 
 
+ARG DEEPDANBOORU="0"
+RUN [[ "${DEEPDANBOORU:-0}" == "0" ]] && : || pip install tensorflow-cpu==2.10 tensorflow-io==0.27.0 git+https://github.com/KichangKim/DeepDanbooru.git@edf73df4cdaeea2cf00e9ac08bd8a9026b7a7b26#egg=deepdanbooru
+
 # Note: don't update the sha of previous versions because the install will take forever
 # instead, update the repo state in a later step
 

services/AUTOMATIC1111/entrypoint.sh

@@ -29,6 +29,7 @@ MOUNTS["${ROOT}/models/SwinIR"]="/data/SwinIR"
 MOUNTS["${ROOT}/models/ScuNET"]="/data/ScuNET"
 MOUNTS["${ROOT}/models/LDSR"]="/data/LDSR"
 MOUNTS["${ROOT}/models/hypernetworks"]="/data/Hypernetworks"
+MOUNTS["${ROOT}/models/deepbooru"]="/data/Deepdanbooru"
 
 MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
 MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"
@@ -58,5 +59,4 @@ if [ -f "/data/config/auto/startup.sh" ]; then
   popd
 fi
 
-
 exec "$@"