Skip to content

GitLab

Unverified Commit ee5be834 authored by Nick Hill's avatar Nick Hill Committed by GitHub
Browse files

[BugFix] Fix 4-GPU RLHF tests (#18007) 


Signed-off-by: default avatarNick Hill <nhill@redhat.com>
parent 48545728
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
.buildkite/test-pipeline.yaml
View file @ ee5be834
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment