Baseten benchmarks

How to install

pip install baseten_benchmarks

For installation without dependencies:

pip install --no-deps baseten_benchmarks

How to run

To hit a local OpenAI server running on post 10001

baseten_benchmark --backend generic \
  --api_url http://localhost:10001/v1/chat/completions \
  --api_key this_does_not_matter \
  --model deepseek \
  --num_prompts 1 2 4 8 16 \
  --concurrency 1 2 4 8 16 \
  --random_input 1024 \
  --output_len 1024 \
  --input_type custom \
  --stream \
  --tokenizer deepseek-ai/DeepSeek-R1 \
  --output_file latency.csv \
  --warmup_requests 2 \
  --prompt_style messages

For now input_type custom uses a fixed text file.

How to publish

poetry config pypi-token.pypi [your pypi token here]
poetry publish --build