diff --git a/docker-compose.yml b/docker-compose.yml
index d4afcfc25..e780c8da7 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 83d41a114..a0d041043 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 "$@"