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

Checking flash gpt2 for starters.

Browse Source
This commit is contained in:
Nicolas Patry 2024-06-05 14:58:02 +02:00
parent 79ab3690aa
commit a55917fb43
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -32,9 +32,11 @@ jobs:
with:
python-version: '3.10'
- uses: actions/checkout@v4
- name: Run tests
- name: install deps
run: |
make install-integration-tests
- name: Run tests
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 }}
python -m pytest -s -vv integration-tests
python -m pytest -s -vv integration-tests/models/test_flash_gpt2.py