Skip to content

Spec dec - #37

Merged
HaibaraAiChan merged 2 commits into
mainfrom
spec_dec
Jan 14, 2026
Merged

Spec dec#37
HaibaraAiChan merged 2 commits into
mainfrom
spec_dec

Conversation

@xiongxu1998

Copy link
Copy Markdown
Collaborator
  1. add tree-based speculative decoding
  2. refactoring the architecture of SD to make it more clear
  3. add simple(Top K) and adaptive(MLP) pruner to reduce the transformation data between multiple devices
  4. currently, batch inference for SD is not support as it's so complicated that I still need some time to debug it.

@HaibaraAiChan
HaibaraAiChan merged commit 68a4c08 into main Jan 14, 2026
1 check passed
@HaibaraAiChan
HaibaraAiChan deleted the spec_dec branch January 14, 2026 01:25
JiuChen0 referenced this pull request in JiuChen0/BloomBee Mar 22, 2026
* 重构pruner代码

* add adaptive pruner

---------

Co-authored-by: Xu Xiong <xiongxu1998@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants