File tree Expand file tree Collapse file tree 3 files changed +6
-10
lines changed
.github/actions/setup-vllm Expand file tree Collapse file tree 3 files changed +6
-10
lines changed Original file line number Diff line number Diff line change @@ -11,13 +11,14 @@ runs:
1111 --name vllm \
1212 -p 8000:8000 \
1313 --privileged=true \
14- quay.io/higginsd/vllm-cpu:65393ee064 \
14+ quay.io/higginsd/vllm-cpu:65393ee064-qwen3 \
1515 --host 0.0.0.0 \
1616 --port 8000 \
1717 --enable-auto-tool-choice \
18- --tool-call-parser llama3_json \
19- --model /root/.cache/Llama-3.2-1B-Instruct \
20- --served-model-name meta-llama/Llama-3.2-1B-Instruct
18+ --tool-call-parser hermes \
19+ --model /root/.cache/Qwen3-0.6B \
20+ --served-model-name Qwen/Qwen3-0.6B \
21+ --max-model-len 8192
2122
2223 # Wait for vllm to be ready
2324 echo "Waiting for vllm to be ready..."
Original file line number Diff line number Diff line change 212212echo " === Running Integration Tests ==="
213213EXCLUDE_TESTS=" builtin_tool or safety_with_image or code_interpreter or test_rag"
214214
215- # Additional exclusions for vllm setup
216- if [[ " $TEST_SETUP " == " vllm" ]]; then
217- EXCLUDE_TESTS=" ${EXCLUDE_TESTS} or test_inference_store_tool_calls"
218- fi
219-
220215PYTEST_PATTERN=" not( $EXCLUDE_TESTS )"
221216if [[ -n " $TEST_PATTERN " ]]; then
222217 PYTEST_PATTERN=" ${PYTEST_PATTERN} and $TEST_PATTERN "
Original file line number Diff line number Diff line change @@ -78,7 +78,7 @@ class Setup(BaseModel):
7878 "VLLM_URL" : "http://localhost:8000/v1" ,
7979 },
8080 defaults = {
81- "text_model" : "vllm/meta-llama/Llama-3.2-1B-Instruct " ,
81+ "text_model" : "vllm/Qwen/Qwen3-0.6B " ,
8282 "embedding_model" : "sentence-transformers/all-MiniLM-L6-v2" ,
8383 },
8484 ),
You can’t perform that action at this time.
0 commit comments