diff --git a/.github/workflows/build-push-vllm-cpu.yml b/.github/workflows/build-push-vllm-cpu.yml index 54e81f9f..1cb8aac4 100644 --- a/.github/workflows/build-push-vllm-cpu.yml +++ b/.github/workflows/build-push-vllm-cpu.yml @@ -39,7 +39,7 @@ jobs: - name: Build and push image run: | IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }} - docker build -f docker/Dockerfile.cpu -t $IMAGE --shm-size=4g . + docker buildx build --platform linux/amd64 -f docker/Dockerfile.cpu -t $IMAGE --shm-size=4g . docker push $IMAGE build_push_arm64_image: @@ -67,5 +67,5 @@ jobs: - name: Build and push image run: | IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }}-arm64 - docker build -f docker/Dockerfile.arm -t $IMAGE --shm-size=4g . + docker buildx build --platform linux/arm64 -f docker/Dockerfile.cpu -t $IMAGE --shm-size=4g . docker push $IMAGE