mlx-examples/llms/mlx_lm/requirements.txt
Awni Hannun 6731254e76
Use fast rope (#945)
* use fast rope

* fix llama

* use fast rope for llama3.1

* requires unreleased mlx

* fix su

* fix deepseek v2

* only one of base or freqs

* nit

* fix

* hard code freqs
2024-08-23 13:18:51 -07:00

7 lines
77 B
Plaintext

mlx>=0.17.0
numpy
transformers[sentencepiece]>=4.39.3
protobuf
pyyaml
jinja2