lmstudio-ai / lmstudio-ai/mlx-engine

Speculative Decoding Reduces tok/sec Speed on M1 Pro Mac

Open
#103 6 comments 0 reactions 0 assignees View on GitHub
enhancement
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.