mlx-examples/llms/speculative_decoding/requirements.txt

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-29 12:51:12 +08:00
mlx>=0.0.6
transformers
numpy
add speculative decoding example for llama (#149) * speculative decoding * add sample 0 * spec decode gives same results as regular decode * rebase * use accept reject criteria * switch to t5 * update readme * readme nit * nits * nits * nits --------- Co-authored-by: Benjamin Anderson <benjamin@Benjamins-MBP.lan> Co-authored-by: Awni Hannun <awni@apple.com> 2023-12-29 07:20:43 +08:00			`mlx>=0.0.6`
			`transformers`
			`numpy`