From efde2563d11fbf9565b67dc6d43715df38982269 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 5 Jun 2024 16:09:07 +0200
Subject: [PATCH] Fix DOCKER_DEVICES.

---
 .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 059d267d..f6f4b693 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -39,5 +39,5 @@ jobs:
         run: |
           export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:2.0.4-rocm
           export HUGGING_FACE_HUB_TOKEN=${{ secrets.HUGGING_FACE_HUB_TOKEN }}
-          export DEVICES=/dev/kfd,/dev/dri
+          export DOCKER_DEVICES=/dev/kfd,/dev/dri
           python -m pytest -s -vv integration-tests/models/test_flash_gpt2.py