fix

huggingface · May 12, 2024 · eaf716f · eaf716f
1 parent 580ec68
commit eaf716f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -77,4 +77,4 @@ jobs:
             --workdir /workspace
           run: |
             pip install -e .[testing,timm,diffusers,codecarbon]
-            pytest -s -x -k 'api and cuda'
+            pytest -s -x -k "api and cuda"
diff --git a/docker/unroot/Dockerfile b/docker/unroot/Dockerfile
@@ -10,9 +10,9 @@ ENV PATH="/home/user/.local/bin:${PATH}"
 RUN addgroup --gid $GROUP_ID group
 RUN adduser --disabled-password --gecos '' --uid $USER_ID --gid $GROUP_ID user
 
-# For ROCm, the user needs to be in the video and render groups
-RUN if [ -d /dev/dri ]; then usermod -a -G video user; fi
-RUN if [ -d /dev/dri ]; then usermod -a -G render user; fi
+# For ROCm, the user needs to be in the video and render groups, check with /opt/rocm/
+RUN if [ -d /opt/rocm/ ]; then usermod -a -G video user; fi
+RUN if [ -d /opt/rocm/ ]; then usermod -a -G render user; fi
 
 USER user
 WORKDIR /home/user