From 909f660f7ced92b73da0cdb7a77460ee40d3da43 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Tue, 21 Nov 2023 15:06:21 +0100 Subject: [PATCH] remoove comment --- .github/workflows/self-scheduled-amd.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/self-scheduled-amd.yml b/.github/workflows/self-scheduled-amd.yml index 95efc7741a7ee4..2faeaa1707e688 100644 --- a/.github/workflows/self-scheduled-amd.yml +++ b/.github/workflows/self-scheduled-amd.yml @@ -348,7 +348,7 @@ jobs: runs-on: [self-hosted, docker-gpu, amd-gpu, '${{ matrix.machine_type }}', '${{ inputs.gpu_flavor }}'] needs: setup container: - image: huggingface/transformers-pytorch-amd-gpu-test # replace with huggingface/transformers-pytorch-deepspeed-amd-gpu + image: huggingface/transformers-pytorch-amd-gpu-test options: --device /dev/kfd --device /dev/dri --env ROCR_VISIBLE_DEVICES --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ steps: - name: Update clone