text-generation-inference/benchmark
yuanwu 8e2e5d8e15 Fix benchmark build error
Signed-off-by: yuanwu <yuan.wu@intel.com>
2024-12-17 05:38:10 +00:00
..
src Fix benchmark build error 2024-12-17 05:38:10 +00:00
Cargo.toml Simplify crossterm imports (#2545) 2024-10-25 09:01:04 +00:00
README.md Update the link to the Ratatui organization (#2546) 2024-10-25 09:01:04 +00:00

Text Generation Inference benchmarking tool

benchmark

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

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