Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "578977bb5ed208c62cf9cff80d955836775e0d24"
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
Show 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: ...@@ -158,8 +158,8 @@ steps:
# TODO: create a dedicated test section for multi-GPU example tests # TODO: create a dedicated test section for multi-GPU example tests
# when we have multiple distributed example tests # when we have multiple distributed example tests
- pushd ../examples/offline_inference - pushd ../examples/offline_inference
- python3 rlhf.py - VLLM_ALLOW_INSECURE_SERIALIZATION=1 python3 rlhf.py
- RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py - VLLM_ALLOW_INSECURE_SERIALIZATION=1 RAY_DEDUP_LOGS=0 python3 rlhf_colocate.py
- popd - popd
- label: Metrics, Tracing Test # 10min - 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