mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
eval max_token_length consistent with training config
instead of hardcoding, follows other envs pattern
This commit is contained in:
parent
bd98a82bbc
commit
5f52befd38
1 changed files with 1 additions and 1 deletions
|
|
@ -259,7 +259,7 @@ class MathEnv(BaseEnv):
|
|||
completion = await managed.completion(
|
||||
prompt=question,
|
||||
n=1,
|
||||
max_tokens=32765,
|
||||
max_tokens=self.config.max_token_length,
|
||||
temperature=0.0,
|
||||
split="eval",
|
||||
stop=stop_list,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue