From 470046519277e9543d9a6d8f102324ff8b44c7db Mon Sep 17 00:00:00 2001
From: Adrien <adrien@huggingface.co>
Date: Mon, 22 Jul 2024 21:50:53 +0200
Subject: [PATCH] use proper name for ci (#2274)

---
 .github/workflows/build.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 41b9e425..89e3f18c 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -28,7 +28,7 @@ jobs:
       group: ${{ github.workflow }}-build-and-push-image-${{ inputs.hardware }}-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
     runs-on:
-      group: aws-r6id-8xlarge-priv
+      group: aws-highmemory-32-plus-priv
     permissions:
       contents: write
       packages: write
@@ -49,7 +49,7 @@ jobs:
                 export dockerfile="Dockerfile"
                 export label_extension=""
                 export docker_devices=""
-                export runs_on="aws-g5-12xlarge"
+                export runs_on="aws-g5-12xlarge-plus"
                 ;;
             rocm)
                 export dockerfile="Dockerfile_amd"