run-cpu-test.sh now runs multi-lora tests

Signed-off-by: Akshat Tripathi <[email protected]>
vllm-project · Jan 10, 2025 · 202aca3 · 202aca3
1 parent 5af9cbb
commit 202aca3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.buildkite/run-cpu-test.sh b/.buildkite/run-cpu-test.sh
@@ -75,6 +75,13 @@ function cpu_tests() {
       --num-prompts 20 \
       --endpoint /v1/completions \
       --tokenizer facebook/opt-125m"
+
+  # Run multi-lora tests
+  docker exec cpu-test-"$BUILDKITE_BUILD_NUMBER"-"$NUMA_NODE" bash -c "
+    set -e
+    pytest -s -v \
+    tests/lora/test_qwen2vl.py \
+    tests/lora/test_lora_bias_e2e.py"
 }
 
 # All of CPU tests are expected to be finished less than 25 mins.