Quellcode durchsuchen

Fix preload for Invoke (#346)

Refs #345
AbdBarho vor 3 Jahren
Ursprung
Commit
aa69f11230
2 geänderte Dateien mit 5 neuen und 5 gelöschten Zeilen
  1. 1 1
      docker-compose.yml
  2. 4 4
      services/invoke/entrypoint.sh

+ 1 - 1
docker-compose.yml

@@ -43,7 +43,7 @@ services:
     <<: *base_service
     <<: *base_service
     profiles: ["invoke"]
     profiles: ["invoke"]
     build: ./services/invoke/
     build: ./services/invoke/
-    image: sd-invoke:25
+    image: sd-invoke:26
     environment:
     environment:
       - PRELOAD=true
       - PRELOAD=true
       - CLI_ARGS=
       - CLI_ARGS=

+ 4 - 4
services/invoke/entrypoint.sh

@@ -38,9 +38,9 @@ for to_path in "${!MOUNTS[@]}"; do
   echo Mounted $(basename "${from_path}")
   echo Mounted $(basename "${from_path}")
 done
 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 "$@"
 exec "$@"