mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 09:21:18 +08:00
|
||
---|---|---|
.. | ||
.gitignore | ||
convert.py | ||
mistral.py | ||
README.md | ||
requirements.txt | ||
test.py |
Mistral
An example of generating text with Mistral using MLX.
Mistral 7B is one of the top large language models in its size class. It is also fully open source with a permissive license1.
Setup
Install the dependencies:
pip install -r requirements.txt
Next, download the model and tokenizer:
curl -O https://files.mistral-7b-v0-1.mistral.ai/mistral-7B-v0.1.tar
tar -xf mistral-7B-v0.1.tar
Then, convert the weights with:
python convert.py
Run
Once you've converted the weights to MLX format, you can generate text with the Mistral model:
python mistral.py --prompt "It is a truth universally acknowledged," --temp 0
Run python mistral.py --help
for more details.
-
Refer to the blog post and github repository for more details. ↩︎