hf_text-generation-inference/benchmark
Nicolas Patry 2cf1f5c00e
Fixing the issue with `add_special_tokens` not being passed around.
2024-08-27 20:06:12 +02:00
..
src Fixing the issue with `add_special_tokens` not being passed around. 2024-08-27 20:06:12 +02:00
Cargo.toml Rebase TRT-llm (#2331) 2024-07-31 10:33:10 +02:00
README.md chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00

README.md

Text Generation Inference benchmarking tool

benchmark

A lightweight benchmarking tool based inspired by oha and powered by tui.

Install

make install-benchmark

Run

First, start text-generation-inference:

text-generation-launcher --model-id bigscience/bloom-560m

Then run the benchmarking tool:

text-generation-benchmark --tokenizer-name bigscience/bloom-560m