Fix preload for Invoke (AbdBarho#346)

Refs AbdBarho#345
Jordan-Lambda · Aug 2, 2023 · a373986 · a373986
1 parent 8f3f7c0
commit a373986
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -43,7 +43,7 @@ services:
     <<: *base_service
     profiles: ["invoke"]
     build: ./services/invoke/
-    image: sd-invoke:25
+    image: sd-invoke:26
     environment:
       - PRELOAD=true
       - CLI_ARGS=

diff --git a/services/invoke/entrypoint.sh b/services/invoke/entrypoint.sh
@@ -38,9 +38,9 @@ for to_path in "${!MOUNTS[@]}"; do
   echo Mounted $(basename "${from_path}")
 done
 
-# if "${PRELOAD}" == "true"; then
-#   set -Eeuo pipefail
-#   python3 -u scripts/preload_models.py --skip-sd-weights --root ${ROOT} --config_file /docker/models.yaml
-# fi
+if "${PRELOAD}" == "true"; then
+  set -Eeuo pipefail
+  invokeai-configure --skip-sd-weights --root ${ROOT} --yes
+fi
 
 exec "$@"