rest

huggingface · Sep 25, 2023 · 7a3ab93 · 7a3ab93
1 parent 33e7f56
commit 7a3ab93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test-amd.yaml b/.github/workflows/test-amd.yaml
@@ -29,7 +29,7 @@ jobs:
     runs-on: [self-hosted, docker-gpu, amd-gpu, multi-amd-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
-      options: --device /dev/kfd --device /dev/dri --shm-size "16gb" -e HIP_VISIBLE_DEVICES=$HIP_VISIBLE_DEVICES -e ROCR_VISIBLE_DEVICES=$ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --device /dev/kfd --device /dev/dri --shm-size "16gb" -e HIP_VISIBLE_DEVICES=$HIP_VISIBLE_DEVICES -e --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
     steps:
       - name: ROCM-SMI
         run: |