Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
hf_text-generation-inference/benchmark
History
OlivierDehaene 895c5f1562
feat(server): only compute prefill logprobs when asked (#406) 
Close #288
 		2023-06-02 17:12:30 +02:00
..
src feat(server): only compute prefill logprobs when asked (#406) 2023-06-02 17:12:30 +02:00
Cargo.toml feat(benchmarker): add summary tables (#368) 2023-05-25 13:38:36 +02:00
README.md feat(benchmark): tui based benchmarking tool (#149) 2023-03-30 15:26:27 +02: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