llama v2 with sharded weights

This commit is contained in:
Awni Hannun
2023-12-12 12:48:15 -08:00
parent 9a02dce35c
commit f0c57c1361
5 changed files with 189 additions and 123 deletions

View File

@@ -1,2 +1,3 @@
mlx
sentencepiece
torch