We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1937e29 commit c3af447Copy full SHA for c3af447
benchmarks/benchmark_serving.py
@@ -17,6 +17,10 @@
17
--dataset-path <path to dataset> \
18
--request-rate <request_rate> \ # By default <request_rate> is inf
19
--num-prompts <num_prompts> # By default <num_prompts> is 1000
20
+
21
+ when using tgi backend, add
22
+ --endpoint /generate_stream
23
+ to the end of the command above.
24
"""
25
import argparse
26
import asyncio
benchmarks/launch_tgi_server.sh
@@ -4,7 +4,7 @@ PORT=8000
4
MODEL=$1
5
TOKENS=$2
6
7
-docker run --gpus all --shm-size 1g -p $PORT:80 \
+docker run -e HF_TOKEN=$HF_TOKEN --gpus all --shm-size 1g -p $PORT:80 \
8
-v $PWD/data:/data \
9
ghcr.io/huggingface/text-generation-inference:1.4.0 \
10
--model-id $MODEL \
0 commit comments