lmstudio-ai / lmstudio-ai/mlx-engine
Speculative Decoding Reduces tok/sec Speed on M1 Pro Mac
- Dominant language
- Python
- Stars
- 1.2k
- Forks
- 133
- Avg merge
- 21h 6m
- Merged PRs (30d)
- 1
Description
Using LMStudio client, I've tried the following combinations:
- Llama 3.1 8B Q4 MLX (main) + Llama 3.2 1B Q4 MLX (draft): speed went from 38 t/s (without draft) to 33.93 (with draft)!
- Same thing except I used Llama 3.2 3B (smaller gap with the main model) Q4 MLX: went from 38.7 t/s to 29.29 t/s
- Qwen 2.5 7B Q4 MLX (main) + Qwen 2.5 1B Q4 MLX (draft): 37.08 t/s to 22.54 t/s
MacBook Pro with M1 Pro Chip, 32GB unified memory. I replicated the same results with GGUF models as well.
Contributor guide
Research direction
Start by reproducing the reported speculative-decoding combinations through the LMStudio client with the listed MLX and GGUF models on an M1 Pro, recording tok/sec with and without a draft model. Done means the cause of the slowdown is identified and the affected combinations no longer regress against the reported baselines.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning, performance
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100