hf_text-generation-inference/benchmark/README.md

31 lines
551 B
Markdown
Raw Normal View History

<div align="center">
# Text Generation Inference benchmarking tool
![benchmark](../assets/benchmark.png)
</div>
2024-02-16 03:58:58 -07:00
A lightweight benchmarking tool based inspired by [oha](https://github.com/hatoo/oha)
and powered by [tui](https://github.com/tui-rs-revival/ratatui).
2024-02-16 03:58:58 -07:00
## Install
2024-02-16 03:58:58 -07:00
```shell
make install-benchmark
```
## Run
First, start `text-generation-inference`:
```shell
text-generation-launcher --model-id bigscience/bloom-560m
```
Then run the benchmarking tool:
```shell
text-generation-benchmark --tokenizer-name bigscience/bloom-560m
2024-02-16 03:58:58 -07:00
```