fix image name

.gitea/workflows/build.yml

@@ -7,12 +7,12 @@ on:
   workflow_dispatch:
 
 env:
-  IMAGE_NAME: spark-vllm
+  IMAGE_NAME: spark-vllm-docker
   IMAGE_TAG: latest
 
 jobs:
   docker:
-    runs-on: nix
+    runs-on: dgx
 
     steps:
       - name: Checkout
@@ -25,29 +25,29 @@ jobs:
           username: ${{ secrets.CI_USER }}
           password: ${{ secrets.CI_PASSWORD }}
 
-      - name: Make build script executable
-        run: chmod +x build-and-copy.sh
 
       - name: Build image using upstream script
+        env:
+          DOCKER_DEFAULT_PLATFORM: linux/arm64
         run: |
-          ./build-and-copy.sh -t ${IMAGE_NAME}:${IMAGE_TAG}
+          bash build-and-copy.sh -t ${IMAGE_NAME}:${IMAGE_TAG}
 
-      - name: Tag image
+      - name: Tag and Push Docker Image
+        shell: bash
         run: |
-          docker tag \
-            ${IMAGE_NAME}:${IMAGE_TAG} \
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${IMAGE_TAG}
+          VERSION=${{ github.run_number }}
+
+          REGISTRY=${GITHUB_SERVER_URL#https://}
+
+          TARGET_IMAGE=$REGISTRY/${{ github.repository_owner }}/${IMAGE_NAME}
 
           docker tag \
             ${IMAGE_NAME}:${IMAGE_TAG} \
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${GITEA_SHA::7}
+            $TARGET_IMAGE:$VERSION
 
-      - name: Push latest
-        run: |
-          docker push \
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${IMAGE_TAG}
+          docker tag \
+            ${IMAGE_NAME}:${IMAGE_TAG} \
+            $TARGET_IMAGE:latest
 
-      - name: Push commit SHA
-        run: |
-          docker push \
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${GITEA_SHA::7}
+          docker push $TARGET_IMAGE:$VERSION
+          docker push $TARGET_IMAGE:latest

build-and-copy.sh

@@ -471,6 +471,7 @@ fi
 COMMON_BUILD_FLAGS+=("--build-arg" "BUILD_JOBS=$BUILD_JOBS")
 COMMON_BUILD_FLAGS+=("--build-arg" "TORCH_CUDA_ARCH_LIST=$GPU_ARCH_LIST")
 COMMON_BUILD_FLAGS+=("--build-arg" "FLASHINFER_CUDA_ARCH_LIST=$GPU_ARCH_LIST")
+COMMON_BUILD_FLAGS+=("--platform" "linux/arm64")
 if [ -n "$NETWORK_ARG" ]; then
     COMMON_BUILD_FLAGS+=("--network" "$NETWORK_ARG")
 fi