Skip to content

Commit 7e893b5

Browse files
lsorberokaris
authored andcommitted
fix: apply missing _convert_text_completion_logprobs_to_chat
1 parent 22b483a commit 7e893b5

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
@@ -3957,7 +3957,9 @@ def chatml_function_calling(
39573957
{
39583958
"finish_reason": "tool_calls",
39593959
"index": 0,
3960-
"logprobs": completion["choices"][0]["logprobs"],
3960+
"logprobs": _convert_text_completion_logprobs_to_chat(
3961+
completion["choices"][0]["logprobs"]
3962+
),
39613963
"message": {
39623964
"role": "assistant",
39633965
"content": None,

0 commit comments

Comments
 (0)