ci/transformers: run tests in tests/generation

Signed-off-by: Dmitry Rogozhkin <[email protected]>
intel · Dec 21, 2024 · 4439570 · 4439570
1 parent 1f28c15
commit 4439570
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.github/workflows/_linux_transformers.yml b/.github/workflows/_linux_transformers.yml
@@ -136,6 +136,16 @@ jobs:
           source activate huggingface_transformers_test
           cd transformers
           python3 -m pytest -rsf --make-reports=$TEST_CASE tests/benchmark || true
+      - name: Run tests/generation
+        env:
+          TEST_CASE: 'tests_generation'
+        run: |
+          source activate huggingface_transformers_test
+          cd transformers
+          # Excluding tests due to:
+          # * torch.distributed.* not yet supported by XPU
+          pattern="not TestFSDPGeneration"
+          python3 -m pytest -rsf --make-reports=$TEST_CASE tests/generation || true
       - name: Run tests/pipelines
         env:
           TEST_CASE: 'tests_pipelines'