fix

huggingface · May 16, 2024 · e1bd44a · e1bd44a
1 parent 86bab66
commit e1bd44a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 16 deletions.
diff --git a/.github/workflows/test_cli_cuda_torch_ort_multi_gpu.yaml b/.github/workflows/test_cli_cuda_torch_ort_multi_gpu.yaml
@@ -36,11 +36,6 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
 
-      - name: Set docker image
-        run: |
-          echo "IMAGE: $REGISTRY/$IMAGE_NAME:latest-$IMAGE_FLAVOR"
-          echo "IMAGE=$REGISTRY/$IMAGE_NAME:latest-$IMAGE_FLAVOR" >> $GITHUB_ENV
-
       - name: Run tests
         uses: addnab/docker-run-action@v3
         with:

diff --git a/.github/workflows/test_cli_cuda_torch_ort_single_gpu.yaml b/.github/workflows/test_cli_cuda_torch_ort_single_gpu.yaml
@@ -36,11 +36,6 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v3
 
-      - name: Set docker image
-        run: |
-          echo "IMAGE: $REGISTRY/$IMAGE_NAME:latest-$IMAGE_FLAVOR"
-          echo "IMAGE=$REGISTRY/$IMAGE_NAME:latest-$IMAGE_FLAVOR" >> $GITHUB_ENV
-
       - name: Run tests
         uses: addnab/docker-run-action@v3
         with:

diff --git a/.github/workflows/test_cli_rocm_pytorch_multi_gpu.yaml b/.github/workflows/test_cli_rocm_pytorch_multi_gpu.yaml
@@ -38,14 +38,12 @@ jobs:
 
       - name: Set target devices
         run: |
-          echo "DEVICE0: $DEVICE0"
-          echo "DEVICE1: $DEVICE1"
+          echo "DEVICE0:$DEVICE0"
+          echo "DEVICE1:$DEVICE1"
           echo "DEVICE0=$DEVICE0" >> $GITHUB_ENV
           echo "DEVICE1=$DEVICE1" >> $GITHUB_ENV
 
       - name: Unroot docker image
-        env:
-          IMAGE:
         run: |
           docker build --build-arg IMAGE=${{ env.IMAGE }} --build-arg USER_ID=$(id -u) --build-arg GROUP_ID=$(id -g) -t ${{ env.IMAGE }}-unroot docker/unroot
 
@@ -54,7 +52,6 @@ jobs:
         env:
           DEVICE0: ${{ env.DEVICE0 }}
           DEVICE1: ${{ env.DEVICE1 }}
-          WORKSPACE: ${{ github.workspace }}
         with:
           image: ${{ env.IMAGE }}-unroot
           options: |
@@ -63,7 +60,7 @@ jobs:
             --device /dev/kfd
             --device /dev/dri/${{ env.DEVICE0 }}
             --device /dev/dri/${{ env.DEVICE1 }}
-            --volume ${{ env.WORKSPACE }}:/workspace
+            --volume ${{ github.workspace }}:/workspace
             --workdir /workspace
           run: |
             pip install -e .[testing,diffusers,timm,deepspeed,peft,autoawq-rocm,auto-gptq-rocm]