Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Bugfix] MLPSpeculator: Use ParallelLMHead in tie_weights=False case. #6303

Merged
merged 1 commit into from
Jul 10, 2024

Conversation

tdoublep
Copy link
Member

@tdoublep tdoublep commented Jul 10, 2024

Fixes #6302

cc @robertgshaw2-neuralmagic

@robertgshaw2-redhat
Copy link
Collaborator

Thanks for the fix!

@mgoin mgoin merged commit c38eba3 into vllm-project:main Jul 10, 2024
70 of 71 checks passed
@tdoublep tdoublep deleted the mlpsepc-notie-fix branch July 10, 2024 13:08
adityagoel14 pushed a commit to adityagoel14/vllm-torchrun-test that referenced this pull request Jul 10, 2024
dtrifiro pushed a commit to opendatahub-io/vllm that referenced this pull request Jul 17, 2024
xjpang pushed a commit to xjpang/vllm that referenced this pull request Jul 24, 2024
Alvant pushed a commit to compressa-ai/vllm that referenced this pull request Oct 26, 2024
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.

[Bug]: MLPSpeculator broken for models with tie_weights=False
3 participants