Skip to content

fix: Strip top_logprobs=null before forwarding to vLLM (#609)#1013

Open
cwing-nvidia wants to merge 1 commit intomainfrom
cwing/fix-top-logprobs-null-crash
Open

fix: Strip top_logprobs=null before forwarding to vLLM (#609)#1013
cwing-nvidia wants to merge 1 commit intomainfrom
cwing/fix-top-logprobs-null-crash

Conversation

@cwing-nvidia
Copy link
Copy Markdown
Contributor

Rollout JSONL files serialized by ng_collect_rollouts include explicit top_logprobs: null, which vLLM interprets differently from omitting the field. With logprobs=True and top_logprobs=null, vLLM does not return logprobs content, causing a TypeError crash in chat_completions.

Rollout JSONL files serialized by ng_collect_rollouts include explicit
top_logprobs: null, which vLLM interprets differently from omitting the
field. With logprobs=True and top_logprobs=null, vLLM does not return
logprobs content, causing a TypeError crash in chat_completions.

Signed-off-by: Chris Wing <cwing@nvidia.com>
Made-with: Cursor
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: Log probabilities are not returned by vLLM when top_logprobs is set to null

2 participants