diff --git a/.github/workflows/3d_parallelism_unit_tests.yaml b/.github/workflows/3d_parallelism_unit_tests.yaml index cd6c2656..66dd6f21 100644 --- a/.github/workflows/3d_parallelism_unit_tests.yaml +++ b/.github/workflows/3d_parallelism_unit_tests.yaml @@ -63,6 +63,7 @@ jobs: --ignore tests/fp8 \ --ignore tests/nanoset \ --verbose \ + --timeout=300 \ tests/ # NOTE: T4 can't run FA2, DoReMi's LLaMa needs FÀ # - name: Run DoReMi tests