mpt-7B-inference/README.md
2023-06-26 01:38:20 -04:00

37 lines
815 B
Markdown

# MPT 30B inference code using CPU
Run inference on the latest MPT-30B model using your CPU.
![Inference Demo]([media//transformers/mpt-30B-inference/media/inference-demo.mp4inference-demo.mp4](https://github.com/abacaj/mpt-30B-inference/assets/7272343/486fc9b1-8216-43cc-93c3-781677235502))
## Requirements
I recommend you use docker for this model, it will make everything easier for you. Tested on cuda-11.8.0 with AMD Epyc CPU.
## Setup
First create a venv.
```sh
python -m venv env && source env/bin/activate
```
Next install dependencies.
```sh
pip install -r requirements.txt
```
Next download the quantized model weights (about 19GB).
```sh
python download_model.py
```
Ready to rock, run inference.
```sh
python inference.py
```
Next modify inference script prompt and generation parameters.