Update 3d_parallelism_unit_tests.yaml

huggingface · Jan 29, 2024 · f675daf · f675daf
1 parent f6d9847
commit f675daf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/3d_parallelism_unit_tests.yaml b/.github/workflows/3d_parallelism_unit_tests.yaml
@@ -59,4 +59,4 @@ jobs:
     - name: Run tests
       # NOTE: -m "not fa2" will run all the unit tests that don't have the mark
       # "fa2" (these are FA2-related tests, we can't run it on T4)
-      run: pytest -m "not fa2" --color=yes --durations=0 --verbose tests/
+      run: pytest -n 5 -m "not fa2" --color=yes --durations=0 --verbose tests/