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

group temps, sample temps, and logprob api params

Browse source
This commit is contained in:
ropresearch 2025-09-25 16:41:58 -04:00
parent efc6b55f0a
commit c3fc68879c
4 changed files with 65 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
atroposlib/envs/base.py
View file

@ -59,6 +59,8 @@ class ScoredDataGroup(TypedDict):
advantages: Optional[List[List[float]]]
ref_logprobs: Optional[List[List[float]]]
messages: Optional[List[List[Message]]]
generation_params: Optional[Dict[str, Any]]
inference_logprobs: Optional[List[List[float]]]
group_overrides: Optional[Dict]
overrides: Optional[List[Dict]]
images: Optional[Any]