Build 5.4.1

.github/workflows/docker-build.yml

@@ -88,7 +88,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "false", invokeai: "4.2.7", python: "3.10", pytorch: "2.2.2", cuda: "12.1.1-base"}
+          - {latest: "false", invokeai: "5.4.1", python: "3.10", pytorch: "2.4.1", cuda: "12.1.1-base"}
 
     steps:
       -

build/COPY_ROOT_99/opt/ai-dock/bin/build/layer99/init.sh

@@ -80,25 +80,33 @@ function build_extra_start() {
 
     # Invoke has no exit/CI run mode so run it and wait until it's fuly initialised
     source "$INVOKEAI_VENV/bin/activate"
-    LD_PRELOAD=libtcmalloc.so invokeai-web 2>&1 | tee /tmp/invoke-ci.log &
+    LD_PRELOAD=libtcmalloc.so invokeai-web 2>&1 | tee /tmp/invoke-ci.log & 
+    INVOKE_PID=$!
+
     wait_max=30
     wait_current=0
     init_string="Uvicorn running on"
 
-    # Until loop to continuously check if the string is found or maximum wait time is reached
     while [ $wait_current -lt $wait_max ]; do
         if grep -qi "$init_string" /tmp/invoke-ci.log; then
             echo "InvokeAI initialization complete."
+            kill -9 $INVOKE_PID
             break
-        else
-            echo "Waiting for InvokeAI initialization to complete..."
-            sleep 1
-            wait_current=$((wait_current + 1))
         fi
+        echo "Waiting for InvokeAI initialization..."
+        sleep 1
+        wait_current=$((wait_current + 1))
     done
 
-    pkill invokeai-web
+    if [ $wait_current -eq $wait_max ]; then
+        echo "Timeout waiting for InvokeAI"
+        kill -9 $INVOKE_PID
+        exit 1
+    fi
+    ps aux
+    echo "Should be exiting about now"
     deactivate
+    echo "Still waiting?"
 }
 
 function build_extra_get_pip_packages() {

docker-compose.yaml

@@ -7,14 +7,22 @@ services:
       context: ./build
       args:
         PYTHON_VERSION: ${PYTHON_VERSION:-3.10}
-        PYTORCH_VERSION: ${PYTORCH_VERSION:-2.2.2}
-        INVOKEAI_VERSION: ${INVOKEAI_VERSION:-4.2.4}
-        IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/python:3.10-v2-cuda-11.8.0-base-22.04}
+        PYTORCH_VERSION: ${PYTORCH_VERSION:-2.4.1}
+        INVOKEAI_VERSION: ${INVOKEAI_VERSION:-5.4.1}
+        IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/python:3.10-v2-cuda-12.1.1-base-22.04}
       tags:
-        - "ghcr.io/ai-dock/invokeai:${IMAGE_TAG:-v2-cuda-11.8.0-base-22.04}"
+        - "ghcr.io/ai-dock/invokeai:${IMAGE_TAG:-v2-cuda-12.1.1-base-22.04}"
 
-    image: ghcr.io/ai-dock/invokeai:${IMAGE_TAG:-v2-cuda-11.8.0-base-22.04}
+    image: ghcr.io/ai-dock/invokeai:${IMAGE_TAG:-v2-cuda-12.1.1-base-22.04}
 
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+
     devices:
       - "/dev/dri:/dev/dri"
       # For AMD GPU