Skip to content

[CB] Add an option to return logprobs#44835

Draft
remi-or wants to merge 22 commits intomainfrom
cb-log-probs
Draft

[CB] Add an option to return logprobs#44835
remi-or wants to merge 22 commits intomainfrom
cb-log-probs

Conversation

@remi-or
Copy link
Collaborator

@remi-or remi-or commented Mar 18, 2026

This PR adds the return_logprobs flag to the continuous batching, enabling the user to retrieve the log probabilites of the tokens generated.
Draft until #44578 is merged

@github-actions
Copy link
Contributor

View the CircleCI Test Summary for this PR:

https://huggingface.co/spaces/transformers-community/circle-ci-viz?pr=44835&sha=09b2d0

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

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