fix image name

.gitea/workflows/build.yml

@@ -7,47 +7,47 @@ 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
         uses: actions/checkout@v4
 
-      - name: Login to Registry
+      - name: Log in to Gitea Container Registry
-        run: |
+        uses: docker/login-action@v3
-          echo "${{ secrets.REGISTRY_PASSWORD }}" | docker login \
+        with:
-            ${{ secrets.REGISTRY_HOST }} \
+          registry: ${{ github.server_url }}
-            -u "${{ secrets.REGISTRY_USERNAME }}" \
+          username: ${{ secrets.CI_USER }}
-            --password-stdin
+          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 \
+          VERSION=${{ github.run_number }}
-            ${IMAGE_NAME}:${IMAGE_TAG} \
+
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${IMAGE_TAG}
+          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
+          docker tag \
-        run: |
+            ${IMAGE_NAME}:${IMAGE_TAG} \
-          docker push \
+            $TARGET_IMAGE:latest
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${IMAGE_TAG}
 
-      - name: Push commit SHA
+          docker push $TARGET_IMAGE:$VERSION
-        run: |
+          docker push $TARGET_IMAGE:latest
-          docker push \
-            ${{ secrets.REGISTRY_HOST }}/${IMAGE_NAME}:${GITEA_SHA::7}

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