Skip to content

Commit

Permalink
[comment] Add link to vLLM logprobs
Browse files Browse the repository at this point in the history
  • Loading branch information
kthui committed Dec 4, 2024
1 parent c38a798 commit 59b1729
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -514,6 +514,7 @@ def _create_response(
)

# logprobs
# https://github.com/vllm-project/vllm/blob/v0.6.3.post1/vllm/sequence.py#L37-L58
if additional_outputs["return_logprobs"]:
if "prev_lens_logprobs" not in request_output_state:
request_output_state["prev_lens_logprobs"] = [0] * len(
Expand Down

0 comments on commit 59b1729

Please sign in to comment.