Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
glegendre01 committed Sep 25, 2023
1 parent df3cf29 commit ea9c482
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/test-amd.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,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" -env HIP_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
steps:
- name: ROCM-SMI
run: |
Expand Down

0 comments on commit ea9c482

Please sign in to comment.