Skip to content

Commit d76084c

Browse files
[ CI ] Re-enable Large Model LM Eval (#6031)
1 parent 80ca1e6 commit d76084c

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -198,6 +198,15 @@ steps:
198198
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
199199
- bash ./run-tests.sh -c configs/models-small.txt -t 1
200200

201+
- label: LM Eval Large Models
202+
gpu: a100
203+
num_gpus: 4
204+
working_dir: "/vllm-workspace/.buildkite/lm-eval-harness"
205+
commands:
206+
- pip install lm-eval
207+
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
208+
- bash ./run-tests.sh -c configs/models-large.txt -t 4
209+
201210
- label: Documentation Build
202211
working_dir: "/vllm-workspace/test_docs/docs"
203212
no_gpu: True

0 commit comments

Comments
 (0)