fix(tgwui): GH action build

Super-Protocol · Nov 26, 2024 · 3406488 · 3406488
1 parent fa3f87f
commit 3406488
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/build-engine.yml b/.github/workflows/build-engine.yml
@@ -51,7 +51,7 @@ jobs:
             if [[ "$SOLUTION" == *cpu ]]; then
               BUILD_ARGS="--build-arg GPU_CHOICE=N"
             elif [[ "$SOLUTION" == *gpu ]]; then
-              BUILD_ARGS="--build-arg GPU_CHOICE=N"
+              BUILD_ARGS="--build-arg GPU_CHOICE=A"
             fi
 
           elif [ "$SOLUTION" == "comfyui" ]; then
@@ -97,7 +97,7 @@ jobs:
       - name: Build Docker image and save to file
         run: |
           IMAGE_FILE="$SOLUTION-image-b${{ github.run_id }}.tar.gz"
-          docker build -t $SOLUTION:latest -f "$DOCKERFILE" "$BUILD_ARGS" "$DOCKERFILE_CONTEXT"
+          docker build -t $SOLUTION:latest -f "$DOCKERFILE" $(echo $BUILD_ARGS) "$DOCKERFILE_CONTEXT"
           docker save $SOLUTION:latest | gzip > $IMAGE_FILE
 
           echo "IMAGE_FILE=$IMAGE_FILE" >> $GITHUB_ENV

diff --git a/Text Generation WebUI/Dockerfile b/Text Generation WebUI/Dockerfile
@@ -1,5 +1,7 @@
 FROM node:20-bookworm AS builder
 
+ARG GPU_CHOICE
+
 WORKDIR /home/node/app
 
 COPY . /home/node/app