From 0de2c559e20cebfaf713d56c18d1a442cc995a91 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 5 Jun 2024 14:42:26 +0200 Subject: [PATCH] CI for amd? --- .github/workflows/build.yaml | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 404605a1..3b0027d7 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -26,7 +26,7 @@ jobs: concurrency: group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }} cancel-in-progress: true - runs-on: intel-pvc-tgi + runs-on: [amd-gpu, mi250] steps: - uses: actions/setup-python@v5 with: @@ -34,7 +34,6 @@ jobs: - name: Run tests run: | make install-integration-tests - export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-a54ca37-intel + 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 }} - docker run $DOCKER_IMAGE --model-id gpt2 python -m pytest -s -vv integration-tests