diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index e7fe1519d..206e71f16 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -29,16 +29,13 @@ jobs: matrix: include: - device: "cuda" - npu_type: "" - - device: "npu" - npu_type: "a2" - - device: "npu" - npu_type: "a3" + - device: "npu-a2" + - device: "npu-a3" runs-on: ubuntu-latest concurrency: - group: ${{ github.workflow }}-${{ github.ref }}-${{ matrix.device }}-${{ matrix.npu_type }} + group: ${{ github.workflow }}-${{ github.ref }}-${{ matrix.device }} cancel-in-progress: ${{ github.ref != 'refs/heads/main' }} environment: @@ -93,7 +90,7 @@ jobs: docker.io/hiyouga/llamafactory:${{ steps.version.outputs.tag }} - name: Build and push Docker image (NPU-A2) - if: ${{ matrix.device == 'npu' && matrix.npu_type == 'a2' }} + if: ${{ matrix.device == 'npu-a2' }} uses: docker/build-push-action@v6 with: context: . @@ -105,7 +102,7 @@ jobs: quay.io/ascend/llamafactory:${{ steps.version.outputs.tag }}-npu-a2 - name: Build and push Docker image (NPU-A3) - if: ${{ matrix.device == 'npu' && matrix.npu_type == 'a3' }} + if: ${{ matrix.device == 'npu-a3' }} uses: docker/build-push-action@v6 with: context: .