mlx-examples/llms/mlx_lm/requirements.txt
Awni Hannun b8a348c1b8
Switch to fast RMS/LN Norm (#603)
* use nn.RMSNorm, use sdpa, cleanup

* bump mlx versions

* minor update

* use fast layer norm

* version bump

* update requirement for whisper

* update requirement for gguf
2024-03-23 07:13:51 -07:00

7 lines
59 B
Plaintext

mlx>=0.8
numpy
transformers>=4.38.0
protobuf
pyyaml
jinja2