From 5a9a5296adc6d11aa4500be96f256b21cd0832a1 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Thu, 23 Nov 2023 19:18:13 +0100
Subject: [PATCH] add 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 2cc44f553f4944..7fe82bda8c9926 100644
--- a/.github/workflows/self-scheduled-amd.yml
+++ b/.github/workflows/self-scheduled-amd.yml
@@ -366,7 +366,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
+      image: huggingface/transformers-pytorch-amd-gpu-test # replace with huggingface/transformers-pytorch-deepspeed-amd-gpu/
       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