Skip to content

Commit 6bfddb6

Browse files
committed
Adapt to latest vllm changes
- Remove --eager-enforce on hpu to improve performance - Refactor to the upstream docker entrypoint changes Fixes issue #631. Signed-off-by: Lianhao Lu <lianhao.lu@intel.com>
1 parent 7219249 commit 6bfddb6

4 files changed

Lines changed: 4 additions & 6 deletions

File tree

helm-charts/common/agent/values.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ tgi:
1414
vllm:
1515
enabled: false
1616
LLM_MODEL_ID: "mistralai/Mistral-7B-Instruct-v0.3"
17-
extraCmdArgs: ["/bin/bash", "-c", "python3 -m vllm.entrypoints.openai.api_server --enforce-eager --model mistralai/Mistral-7B-Instruct-v0.3 --tensor-parallel-size 1 --host 0.0.0.0 --port 2080 --download-dir /data --block-size 128 --max-num-seqs 4096 --max-seq_len-to-capture 8192 --enable-auto-tool-choice --tool-call-parser mistral"]
17+
extraCmdArgs: ["--tensor-parallel-size", "1", "--block-size", "128", "--max-num-seqs", "4096", "--max-seq_len-to-capture", "8192", "--enable-auto-tool-choice", "--tool-call-parser", "mistral"]
1818

1919
replicaCount: 1
2020
llm_endpoint_url: ""

helm-charts/common/llm-uservice/ci-vllm-gaudi-values.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ vllm:
1313
tag: "latest"
1414
LLM_MODEL_ID: Intel/neural-chat-7b-v3-3
1515
OMPI_MCA_btl_vader_single_copy_mechanism: none
16-
extraCmdArgs: ["--enforce-eager","--tensor-parallel-size","1","--block-size","128","--max-num-seqs","256","--max-seq_len-to-capture","2048"]
16+
extraCmdArgs: ["--tensor-parallel-size","1","--block-size","128","--max-num-seqs","256","--max-seq_len-to-capture","2048"]
1717
resources:
1818
limits:
1919
habana.ai/gaudi: 1

helm-charts/common/vllm/gaudi-values.yaml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,7 @@ image:
1111

1212
# VLLM_CPU_KVCACHE_SPACE: "40"
1313
OMPI_MCA_btl_vader_single_copy_mechanism: none
14-
extraCmdArgs: ["--enforce-eager","--tensor-parallel-size","1","--block-size","128","--max-num-seqs","256","--max-seq_len-to-capture","2048"]
15-
# Workaround for current HPU image with start command /bin/bash
16-
# extraCmdArgs: ["/bin/bash","-c","python3 -m vllm.entrypoints.openai.api_server --enforce-eager --model Intel/neural-chat-7b-v3-3 --tensor-parallel-size 1 --host 0.0.0.0 --port 2080 --download-dir /data --block-size 128 --max-num-seqs 256 --max-seq_len-to-capture 2048"]
14+
extraCmdArgs: ["--tensor-parallel-size","1","--block-size","128","--max-num-seqs","256","--max-seq_len-to-capture","2048"]
1715
resources:
1816
limits:
1917
habana.ai/gaudi: 1

helm-charts/common/vllm/values.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ resources: {}
5151
# cpu: 100m
5252
# memory: 128Mi
5353

54-
extraCmdArgs: ["--enforce-eager", "--dtype", "auto"]
54+
extraCmdArgs: []
5555

5656
livenessProbe:
5757
httpGet:

0 commit comments

Comments
 (0)