Custom Models Auto (#75)

cache/.gitignore

@@ -2,3 +2,4 @@
 /transformers
 /weights
 /models
+/custom-models

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}

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
 }

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

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