From 67c0ff67ee8771c4ab954e02a36170c16d7f6120 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 2 Oct 2024 17:42:07 +0200
Subject: [PATCH] Without pytest.

---
 .github/workflows/build.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 2b13962f..d035b3a6 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -191,7 +191,7 @@ jobs:
         run: |
           export DOCKER_VOLUME=/mnt2/
           # export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
-          export DOCKER_IMAGE=busybox
+          export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-318bbc7-rocm
           # export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
           export DOCKER_DEVICES=/dev/kfd,/dev/cri
           # export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}"