From d0a7eeb84aeb4edd535f21028f4b09134497ba7b Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Mon, 12 May 2025 10:46:52 -0700 Subject: [PATCH] [BugFix] Fix 4-GPU RLHF tests These require the recently added VLLM_ALLOW_INSECURE_SERIALIZATION env var to be set. Signed-off-by: Nick Hill --- .buildkite/test-pipeline.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index 027cb218df5e..7f43c535428b 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -158,8 +158,8 @@ steps: # TODO: create a dedicated test section for multi-GPU example tests # when we have multiple distributed example tests - pushd ../examples/offline_inference - - python3 rlhf.py - - RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py + - VLLM_ALLOW_INSECURE_SERIALIZATION=1 python3 rlhf.py + - VLLM_ALLOW_INSECURE_SERIALIZATION=1 RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py - popd - label: Metrics, Tracing Test # 10min