diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml index c5f56eed5..c34f47e61 100644 --- a/.github/workflows/unit_tests.yml +++ b/.github/workflows/unit_tests.yml @@ -61,7 +61,7 @@ env: PYTORCH_ALLOC_CONF: 'expandable_segments:True' MAX_JOBS: 4 RUNNER: 10.0.13.31 - XEON5: 10.0.14.248 + XEON5: 10.0.14.249 CUDA_VERSION: 128 TORCH_VERSION: 2.8.0 PYTHON_VERSION: 312 # use 312, no need to recompile libs like flash_attn...