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>
This commit is contained in:
Vaibhav Srivastav
2023-12-23 03:40:25 +05:30
committed by GitHub
parent 7ae445f6c7
commit 0eaa323c10
4 changed files with 6 additions and 26 deletions

View File

@@ -1,3 +1,3 @@
mlx
mlx>=0.0.6
sentencepiece
torch