NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-28 17:29:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-03-13 20:13:30 +00:00
parent a8cdb53a4d
commit 82964b6e48
2 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
atroposlib/envs/teacher_distillation_env.py
View file

@ -175,9 +175,7 @@ class TeacherDistillationEnv(BaseEnv, ABC):
group["distill_logprobs"] = None
return group
top_k = int(
group_overrides.get("teacher_top_k", self.config.teacher_top_k)
)
top_k = int(group_overrides.get("teacher_top_k", self.config.teacher_top_k))
if top_k <= -1:
group["distill_token_ids"] = None
group["distill_logprobs"] = None