diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 86a96370..404605a1 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -36,4 +36,5 @@ jobs: make install-integration-tests export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-a54ca37-intel export HUGGING_FACE_HUB_TOKEN=${{ secrets.HUGGING_FACE_HUB_TOKEN }} + docker run $DOCKER_IMAGE --model-id gpt2 python -m pytest -s -vv integration-tests