README.md 551 Bytes
Newer Older
1
2
3
4
5
6
7
8
<div align="center">

# Text Generation Inference benchmarking tool

![benchmark](../assets/benchmark.png)

</div>

OlivierDehaene's avatar
OlivierDehaene committed
9
A lightweight benchmarking tool based inspired by [oha](https://github.com/hatoo/oha)
10
11
and powered by [tui](https://github.com/tui-rs-revival/ratatui).

OlivierDehaene's avatar
OlivierDehaene committed
12
## Install
13

OlivierDehaene's avatar
OlivierDehaene committed
14
```shell
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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
OlivierDehaene's avatar
OlivierDehaene committed
30
```