mlx-examples/llms/mlx_lm/requirements.txt
Awni Hannun 8b05bb6d18
[mlx-lm] Use sdpa in llama / mistral model (#515)
* use sdpa

* update a few more models

* version

* fix stablelm type
2024-03-07 17:41:23 -08:00

6 lines
52 B
Plaintext

mlx>=0.6
numpy
transformers>=4.38.0
protobuf
pyyaml