slow completions over group_size 4, debugging added

This commit is contained in:
Shannon Sands 2026-02-05 10:57:13 +10:00
parent 5b82190460
commit 88286f6da3
3 changed files with 22 additions and 5 deletions

View file

@ -448,6 +448,8 @@ class AgentEnv(BaseEnv, ABC, Generic[AgentEnvConfigT]):
if len(items) != self.config.group_size:
return None, backlog
# TODO: Mack sure logprobs included
group: ScoredDataGroup = ScoredDataGroup(
tokens=[],