Update CI.yml

.github/workflows/CI.yml

@@ -27,7 +27,7 @@ jobs:
         platform:
           - runner: ubuntu-latest
             target: x86_64
-          - runner: ubuntu-24.04-arm
+          - runner: ubuntu-latest
             target: aarch64
     steps:
       - uses: actions/checkout@v4
@@ -130,7 +130,7 @@ jobs:
           args: --non-interactive --skip-existing wheels-*/*
 
   push-docker:
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.platform }}
     if: "startsWith(github.ref, 'refs/tags/')"
     permissions:
       contents: read
@@ -138,6 +138,7 @@ jobs:
     strategy:
       matrix:
         tag: [cpu, cuda]
+        platform: [ubuntu-latest, ubuntu-24.04-arm]
     steps:
       - uses: actions/checkout@v4
       - name: Set up QEMU
@@ -158,4 +159,3 @@ jobs:
           tags: |
             ghcr.io/${{ github.repository }}:${{ matrix.tag }}
           file: docker/${{ matrix.tag }}.Dockerfile
-          platforms: linux/amd64, linux/arm64