# models.TransformerWorkload { #mlsysim.models.TransformerWorkload }

```python
models.TransformerWorkload()
```



## Methods

| Name | Description |
| --- | --- |
| [get_kv_cache_size](#mlsysim.models.TransformerWorkload.get_kv_cache_size) | Calculates memory footprint for the KV cache. |

### get_kv_cache_size { #mlsysim.models.TransformerWorkload.get_kv_cache_size }

```python
models.TransformerWorkload.get_kv_cache_size(
    seq_len,
    batch_size,
    precision=BYTES_FP16,
)
```

Calculates memory footprint for the KV cache.
