Skip to content

[Model Runner V2] Fix processed logits in sample()#37144

Merged
WoosukKwon merged 1 commit intomainfrom
woosuk/mrv2-fix-logits
Mar 16, 2026
Merged

[Model Runner V2] Fix processed logits in sample()#37144
WoosukKwon merged 1 commit intomainfrom
woosuk/mrv2-fix-logits

Conversation

@WoosukKwon
Copy link
Copy Markdown
Collaborator

Part of #36930. Split into a separate PR for faster merge

Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
@WoosukKwon WoosukKwon requested a review from njhill as a code owner March 16, 2026 06:17
@WoosukKwon WoosukKwon added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 16, 2026
@mergify mergify bot added the v1 label Mar 16, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request corrects a bug in the sample method within the V2 GPU sampler. The method was previously returning the original, unprocessed logits instead of the processed_logits that result from applying sampling parameters. This change ensures the correct processed_logits are returned, which is crucial for accurate log probability calculations when logprobs_mode is set to processed_logprobs. The fix is correct and directly addresses the issue.

@WoosukKwon WoosukKwon merged commit 96efb91 into main Mar 16, 2026
56 of 58 checks passed
@WoosukKwon WoosukKwon deleted the woosuk/mrv2-fix-logits branch March 16, 2026 07:35
Lucaskabela pushed a commit to Lucaskabela/vllm that referenced this pull request Mar 17, 2026
wendyliu235 pushed a commit to wendyliu235/vllm-public that referenced this pull request Mar 18, 2026
fxdawnn pushed a commit to fxdawnn/vllm that referenced this pull request Mar 19, 2026
khairulkabir1661 pushed a commit to khairulkabir1661/vllm that referenced this pull request Mar 27, 2026
Monishver11 pushed a commit to Monishver11/vllm that referenced this pull request Mar 27, 2026
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
Signed-off-by: Monishver Chandrasekaran <monishverchandrasekaran@gmail.com>
JiantaoXu pushed a commit to JiantaoXu/vllm that referenced this pull request Mar 28, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed v1

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant