Skip to content

Return logprobs in delayed_sampling #1323

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

Merged
merged 8 commits into from
May 29, 2025

Conversation

kdamaszk
Copy link

@kdamaszk kdamaszk commented May 27, 2025

Currently when using delayed_sampling we are not updating logprobs. It means that if user will request them through SamplingParams, he will receive None.
This PR fixes that by adding calculation of logprobs to sampler postprocessing.

Signed-off-by: Karol Damaszke <kdamaszke@habana.ai>
kdamaszk added 6 commits May 27, 2025 16:55
Signed-off-by: Karol Damaszke <kdamaszke@habana.ai>
Signed-off-by: Karol Damaszke <kdamaszke@habana.ai>
Signed-off-by: Karol Damaszke <kdamaszke@habana.ai>
Signed-off-by: Karol Damaszke <kdamaszke@habana.ai>
@kdamaszk
Copy link
Author

/run-gaudi-tests

Signed-off-by: Karol Damaszke <kdamaszke@habana.ai>
@kdamaszk
Copy link
Author

/run-gaudi-tests

1 similar comment
@madamczyk-intel
Copy link

/run-gaudi-tests

@kdamaszk kdamaszk enabled auto-merge (squash) May 29, 2025 12:14
@kdamaszk kdamaszk merged commit 4ec0826 into habana_main May 29, 2025
48 of 49 checks passed
@kdamaszk kdamaszk deleted the dev/kdamaszke/add-logprobs-to-cached-output branch May 29, 2025 12:44
ccrhx4 pushed a commit to ccrhx4/huanxing.vllm-fork that referenced this pull request Jul 16, 2025
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.

3 participants