From a55917fb43cfd4b6da1716bc9f9e2cc0928153da Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 5 Jun 2024 14:58:02 +0200
Subject: [PATCH] Checking flash gpt2 for starters.

---
 .github/workflows/build.yaml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index c194aafe..f60a2f59 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -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