From 97ff6130b86d5a8015cc358f8ef283aaef48bbc8 Mon Sep 17 00:00:00 2001
From: Pauline Bailly-Masson <155966238+paulinebm@users.noreply.github.com>
Date: Mon, 15 Apr 2024 15:09:06 +0200
Subject: [PATCH] Update test-pytorch-xla-tpu-tgi.yml

---
 .github/workflows/test-pytorch-xla-tpu-tgi.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
index 4af21f44..411f60da 100644
--- a/.github/workflows/test-pytorch-xla-tpu-tgi.yml
+++ b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -17,7 +17,7 @@ concurrency:
 jobs:
   do-the-job:
     name: Run TGI tests
-    runs-on: optimum-tpu
+    runs-on: tpu-runners-dind
     container:
       # Use an image that works with TPU with Pytorch 2.3.0 (release was not working)
       image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla@sha256:8f1dcd5b03f993e4da5c20d17c77aff6a5f22d5455f8eb042d2e4b16ac460526