## Building

Select one of the following models to build:

- `gpt-neo-1.3B`
- `gpt-neo-2.7B`
- `gpt-neox-20b`
- `pygmalion-6b`
- `gpt-j-6b`

```BASH
docker build --build-arg MODEL_NAME={model name} -t repo/image_name:tag .
```