improve build ci

Signed-off-by: Adrien <adrien@huggingface.co>
This commit is contained in:
Adrien 2024-07-22 14:34:45 +02:00
parent 56e7f5d779
commit 1c8b78ae99
No known key found for this signature in database
GPG Key ID: 13A249A288DE0DAB

View File

@ -49,7 +49,7 @@ jobs:
export dockerfile="Dockerfile"
export label_extension=""
export docker_devices=""
export runs_on="aws-g6-12xlarge"
export runs_on="aws-g5-12xlarge"
;;
rocm)
export dockerfile="Dockerfile_amd"