From aa69f112302d29d667f20c9ccbe243c10834f6a2 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Mon, 27 Feb 2023 19:59:36 +0100
Subject: [PATCH] Fix preload for Invoke (#346)

Refs #345
---
 docker-compose.yml            | 2 +-
 services/invoke/entrypoint.sh | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index d4afcfc..e780c8d 100644
--- 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
index 83d41a1..a0d0410 100755
--- 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 "$@"