NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

set prompt logprobs to a masked value

Browse source
This commit is contained in:
dmahan93 2025-10-26 11:58:55 -07:00
parent c22f8ca81b
commit c483840f59
2 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
atroposlib/envs/server_handling/managed_server.py
View file

@ -243,7 +243,7 @@ class ManagedServer:
elif len(output_logprobs) > len(output_tokens):
output_logprobs = output_logprobs[: len(output_tokens)]
full_logprobs = [0.0] * prompt_len + output_logprobs
full_logprobs = [1.0] * prompt_len + output_logprobs
return SequenceNode(
full_text=full_text,