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

change to 128 tokens to support low length rejection

Browse source
This commit is contained in:
Dakota 2026-02-04 16:23:30 -06:00
parent 10f651289c
commit 9ff24bf370
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -519,9 +519,9 @@ class DummyManagedServer:
"""
# Fixed dummy values
DUMMY_TOKENS = [1, 2, 3]
DUMMY_MASKED_TOKENS = [-100, 2, 3]
DUMMY_LOGPROBS = [-0.5, -0.5, -0.5]
DUMMY_TOKENS = [i for i in range(128)]
DUMMY_MASKED_TOKENS = [-100] + DUMMY_TOKENS[1:]
DUMMY_LOGPROBS = [-0.5 for _ in range(128)]
def __init__(
self,