From 59892da8661054f522042a10dd6d991f05803561 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Sat, 17 Sep 2022 13:44:00 +0200
Subject: [PATCH] Custom Models Auto (#75)

---
 cache/.gitignore                   |  1 +
 services/AUTOMATIC1111/Dockerfile  |  5 +++--
 services/AUTOMATIC1111/config.json | 12 +++++++++++-
 services/AUTOMATIC1111/mount.sh    |  3 +--
 services/download/download.sh      |  2 ++
 5 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/cache/.gitignore b/cache/.gitignore
index 0988bd2e8..a2aafd457 100644
--- a/cache/.gitignore
+++ b/cache/.gitignore
@@ -2,3 +2,4 @@
 /transformers
 /weights
 /models
+/custom-models
diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile
index 3b8e03530..317231c9e 100644
--- a/services/AUTOMATIC1111/Dockerfile
+++ b/services/AUTOMATIC1111/Dockerfile
@@ -42,7 +42,7 @@ RUN pip install --prefer-binary --no-cache-dir -r ${ROOT}/repositories/CodeForme
 # Note: don't update the sha of previous versions because the install will take forever
 # instead, update the repo state in a later step
 
-ARG SHA=7fe00d08402b8bf9f7f0ffef59ee3f3ad0187cfc
+ARG SHA=99585b3514e2d7e987651d5c6a0806f933af012b
 RUN <<EOF
 cd stable-diffusion-webui
 git pull --rebase
@@ -61,4 +61,5 @@ RUN chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py
 WORKDIR ${WORKDIR}
 EXPOSE 7860
 # run, -u to not buffer stdout / stderr
-CMD /docker/mount.sh && python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config ${CLI_ARGS}
+CMD /docker/mount.sh && \
+  python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config --ckpt-dir /cache/custom-models --ckpt /cache/models/model.ckpt ${CLI_ARGS}
diff --git a/services/AUTOMATIC1111/config.json b/services/AUTOMATIC1111/config.json
index 544257f6d..ac23639ad 100644
--- a/services/AUTOMATIC1111/config.json
+++ b/services/AUTOMATIC1111/config.json
@@ -44,5 +44,15 @@
   "interrogate_clip_num_beams": 1,
   "interrogate_clip_min_length": 24,
   "interrogate_clip_max_length": 48,
-  "interrogate_clip_dict_limit": 1500.0
+  "interrogate_clip_dict_limit": 1500.0,
+  "samples_filename_pattern": "",
+  "directories_filename_pattern": "",
+  "save_selected_only": false,
+  "filter_nsfw": false,
+  "img2img_color_correction": false,
+  "img2img_fix_steps": false,
+  "enable_quantization": false,
+  "enable_batch_seeds": true,
+  "memmon_poll_rate": 8,
+  "sd_model_checkpoint": null
 }
diff --git a/services/AUTOMATIC1111/mount.sh b/services/AUTOMATIC1111/mount.sh
index 4833676c8..396811099 100755
--- a/services/AUTOMATIC1111/mount.sh
+++ b/services/AUTOMATIC1111/mount.sh
@@ -32,5 +32,4 @@ rm -rf ${ROOT}/repositories/CodeFormer/weights/CodeFormer ${ROOT}/repositories/C
 ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/CodeFormer
 ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/facelib
 
-# mount config
-# ln -sf /docker/config.json ${WORKDIR}/config.json
+mkdir -p /cache/torch /cache/transformers /cache/weights /cache/models /cache/custom-models
diff --git a/services/download/download.sh b/services/download/download.sh
index 35048c02a..d85e3a492 100755
--- a/services/download/download.sh
+++ b/services/download/download.sh
@@ -2,6 +2,8 @@
 
 set -Eeuo pipefail
 
+mkdir -p /cache/torch /cache/transformers /cache/weights /cache/models /cache/custom-models
+
 echo "Downloading, this might take a while..."
 
 aria2c --input-file /docker/links.txt --dir /cache/models --continue