Skip to content

Commit ee5be83

Browse files
authored
[BugFix] Fix 4-GPU RLHF tests (#18007)
Signed-off-by: Nick Hill <nhill@redhat.com>
1 parent 4854572 commit ee5be83

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,8 +158,8 @@ steps:
158158
# TODO: create a dedicated test section for multi-GPU example tests
159159
# when we have multiple distributed example tests
160160
- pushd ../examples/offline_inference
161-
- python3 rlhf.py
162-
- RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py
161+
- VLLM_ALLOW_INSECURE_SERIALIZATION=1 python3 rlhf.py
162+
- VLLM_ALLOW_INSECURE_SERIALIZATION=1 RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py
163163
- popd
164164

165165
- label: Metrics, Tracing Test # 10min

0 commit comments

Comments
 (0)