Skip to content

GitLab

Unverified Commit 3453b964 authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[Misc][Doc] Minor benchmark README update (#14874) 


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent 61c6a5a7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
benchmarks/README.md
View file @ 3453b964
...@@ -82,10 +82,10 @@ Then run the benchmarking script ...@@ -82,10 +82,10 @@ Then run the benchmarking script
# wget https://huggingface.co/datasets/anon8231489123/ShareGPT_Vicuna_unfiltered/resolve/main/ShareGPT_V3_unfiltered_cleaned_split.json # wget https://huggingface.co/datasets/anon8231489123/ShareGPT_Vicuna_unfiltered/resolve/main/ShareGPT_V3_unfiltered_cleaned_split.json
MODEL_NAME="NousResearch/Hermes-3-Llama-3.1-8B" MODEL_NAME="NousResearch/Hermes-3-Llama-3.1-8B"
NUM_PROMPTS=10 NUM_PROMPTS=10
BACKEND="openai-chat" BACKEND="vllm"
DATASET_NAME="sharegpt" DATASET_NAME="sharegpt"
DATASET_PATH="<your data path>/ShareGPT_V3_unfiltered_cleaned_split.json" DATASET_PATH="<your data path>/ShareGPT_V3_unfiltered_cleaned_split.json"
python3 vllm/benchmarks/benchmark_serving.py --backend ${BACKEND} --model ${MODEL_NAME} --endpoint /v1/chat/completions --dataset-name ${DATASET_NAME} --dataset-path ${DATASET_PATH} --num-prompts ${NUM_PROMPTS} python3 vllm/benchmarks/benchmark_serving.py --backend ${BACKEND} --model ${MODEL_NAME} --endpoint /v1/completions --dataset-name ${DATASET_NAME} --dataset-path ${DATASET_PATH} --num-prompts ${NUM_PROMPTS}
``` ```
If successful, you will see the following output If successful, you will see the following output
......
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