mlx-examples/llms/mistral/requirements.txt
Vaibhav Srivastav 0eaa323c10
Fix conversion + inference errors. - Mistral (#176)
* Fix conversion + inference errors.

* wire rope_theta throuugh to nn.RoPE

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-22 14:10:25 -08:00

5 lines
37 B
Plaintext

mlx>=0.0.6
sentencepiece
torch
numpy