huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

CI for amd?

Browse Source
This commit is contained in:
Nicolas Patry 2024-06-05 14:42:26 +02:00
parent 7634bed00b
commit 0de2c559e2
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/build.yaml vendored
View File

@ -26,7 +26,7 @@ jobs:
concurrency: concurrency:
group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }} group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true cancel-in-progress: true
runs-on: intel-pvc-tgi runs-on: [amd-gpu, mi250]
steps: steps:
- uses: actions/setup-python@v5 - uses: actions/setup-python@v5
with: with:
@ -34,7 +34,6 @@ jobs:
- name: Run tests - name: Run tests
run: | run: |
make install-integration-tests 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 }} export HUGGING_FACE_HUB_TOKEN=${{ secrets.HUGGING_FACE_HUB_TOKEN }}
docker run $DOCKER_IMAGE --model-id gpt2
python -m pytest -s -vv integration-tests python -m pytest -s -vv integration-tests