From eec6c3241b34bf4284d4eef4e32d073a486b7fd3 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Sat, 8 Jun 2024 21:55:27 +0200
Subject: [PATCH] .

---
 .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 f3c67e9153e..ca5e4755c85 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -48,15 +48,14 @@ jobs:
                 export dockerfile="Dockerfile"
                 export label_extension=""
                 export docker_devices=""
-                # export runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
                 export runs_on="nvidia-gpu"
                 ;;
             rocm)
                 export dockerfile="Dockerfile_amd"
                 export label_extension="-rocm"
                 export docker_devices="/dev/kfd,/dev/dri"
-                # export runs_on="[self-hosted, amd-gpu , multi-gpu]"
                 export runs_on="amd-gpu-tgi"
+                # export runs_on="ubuntu-latest"
                 ;;
             intel)
                 export dockerfile="Dockerfile_intel"
@@ -168,6 +167,7 @@ jobs:
           make install-integration-tests
       - name: Tailscale
         uses: huggingface/tailscale-action@main
+        if: needs.build-and-push.outputs.runs_on != 'amd-gpu-tgi'
         with:
           authkey: ${{ secrets.TAILSCALE_AUTHKEY }}
       - name: Run tests