Currently, Distributed Llama supports three types of Hugging Face models: llama
, mistral
, and mixtral
. You can try to convert any compatible Hugging Face model and run it with Distributed Llama.
Important
All converters are in the early stages of development. After conversion, the model may not work correctly.
- Download a model, for example: Mistral-7B-v0.3.
- The downloaded model should contain
config.json
,tokenizer.json
,tokenizer_config.json
andtokenizer.model
and safetensor files. - Run the converter of the model:
cd converter
python convert-hf.py path/to/hf/model q40 mistral-7b-0.3
- Run the converter of the tokenizer:
python convert-tokenizer-hf.py path/to/hf/model mistral-7b-0.3
- That's it! Now you can run the Distributed Llama.
./dllama inference --model dllama_model_mistral-7b-0.3_q40.m --tokenizer dllama_tokenizer_mistral-7b-0.3.t --buffer-float-type q80 --prompt "Hello world"