Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit f62ded4

Browse files
lsorberXyLearningProgramming
authored andcommitted
fix: apply missing _convert_text_completion_logprobs_to_chat
1 parent ffed3ed commit f62ded4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

llama_cpp/llama_chat_format.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3955,7 +3955,9 @@ def chatml_function_calling(
39553955
{
39563956
"finish_reason": "tool_calls",
39573957
"index": 0,
3958-
"logprobs": completion["choices"][0]["logprobs"],
3958+
"logprobs": _convert_text_completion_logprobs_to_chat(
3959+
completion["choices"][0]["logprobs"]
3960+
),
39593961
"message": {
39603962
"role": "assistant",
39613963
"content": None,

0 commit comments

Comments
 (0)