Build multi-arch image

infiniflow · Oct 3, 2024 · 9537953 · 9537953
1 parent 0082083
commit 9537953
Showing 1 changed file with 21 additions and 4 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -45,15 +45,32 @@ jobs:
       - name: Check out code
         uses: actions/checkout@v4
 
-      - name: Build ragflow:dev-slim
+      - name: Prepare data
         run: |
           RUNNER_WORKSPACE_PREFIX=${RUNNER_WORKSPACE_PREFIX:-$HOME}
           cp -r ${RUNNER_WORKSPACE_PREFIX}/huggingface.co ${RUNNER_WORKSPACE_PREFIX}/nltk_data .
-          sudo docker build -f Dockerfile.slim -t infiniflow/ragflow:dev-slim .
+
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build ragflow:dev-slim
+        uses: docker/build-push-action@v6
+        with:
+          platforms: linux/amd64,linux/arm64
+          push: false
+          tags: infiniflow/ragflow:dev-slim
+          file: Dockerfile.slim
 
       - name: Build ragflow:dev
-        run: |
-          sudo docker build -f Dockerfile -t infiniflow/ragflow:dev .
+        uses: docker/build-push-action@v6
+        with:
+          platforms: linux/amd64,linux/arm64
+          push: false
+          tags: infiniflow/ragflow:dev
+          file: Dockerfile
 
       - name: Start ragflow:dev-slim
         run: |