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

added missing logprob

Browse source
This commit is contained in:
kit 2025-12-24 00:00:34 +00:00
parent 5e962e682e
commit 6eb2e49618
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
environments/gsm8k_server.py
View file

@ -272,6 +272,7 @@ class GSM8kEnv(BaseEnv):
scores["tokens"] = list()
scores["masks"] = list()
scores["scores"] = list()
scores["inference_logprobs"] = list()
gold_parsed = parse(
rollout_group_data[0]["gold_answer"],
extraction_mode="first_match",