Run inference on MPT-7B using CPU

Find a file

Anton Bacaj 3bcb893fd5 Update README.md		2023-06-26 01:38:20 -04:00
media	initial commit	2023-06-26 05:36:27 +00:00
.gitignore	Initial commit	2023-06-26 01:30:01 -04:00
download_model.py	initial commit	2023-06-26 05:36:27 +00:00
inference.py	initial commit	2023-06-26 05:36:27 +00:00
LICENSE	Initial commit	2023-06-26 01:30:01 -04:00
README.md	Update README.md	2023-06-26 01:38:20 -04:00
requirements.txt	initial commit	2023-06-26 05:36:27 +00:00

MPT 30B inference code using CPU

Run inference on the latest MPT-30B model using your CPU.

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.

First create a venv.

python -m venv env && source env/bin/activate

Next install dependencies.

pip install -r requirements.txt

Next download the quantized model weights (about 19GB).

python download_model.py

Ready to rock, run inference.

python inference.py

Next modify inference script prompt and generation parameters.