mlx-examples/llms/deepseek-coder/requirements.txt

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-29 04:31:13 +08:00
torch
mlx
numpy
transformers>=4.35
add deepseek coder example (#172) * 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> 2023-12-29 13:42:22 +08:00			`torch`
			`mlx`
			`numpy`
			`transformers>=4.35`