zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-16 02:08:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add deepseek coder example (#172)

Browse Source 
* feat: add example for deepseek coder

* chore: remove hardcoded rope_scaling_factor

* feat: add quantization support

* chore: update readme

* chore: clean up the rope scalling factor param in create cos sin theta

* feat: add repetition_penalty

* style /consistency changes to ease future integration

* nits in README

* one more typo

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
Anchen
2023-12-29 16:42:22 +11:00
committed by GitHub
parent 37fd2464dc
commit 31ddbd7806
9 changed files with 529 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
llms/deepseek-coder/requirements.txt Normal file
View File

@@ -0,0 +1,4 @@
torch
mlx
numpy
transformers>=4.35