NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-05-01 17:45:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

eval max_token_length consistent with training config

Browse source 
instead of hardcoding, follows other envs pattern
This commit is contained in:
Partho Das 2026-03-03 18:03:04 +05:30
parent bd98a82bbc
commit 5f52befd38
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
environments/math_server_zero.py
View file

@ -259,7 +259,7 @@ class MathEnv(BaseEnv):
completion = await managed.completion( completion = await managed.completion(
prompt=question, prompt=question,
n=1, n=1,
max_tokens=32765, max_tokens=self.config.max_token_length,
temperature=0.0, temperature=0.0,
split="eval", split="eval",
stop=stop_list, stop=stop_list,