NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-22 16:48:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

proper fallback

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-20 01:22:50 -05:00
parent 3910a58f9b
commit 559d649a26
3 changed files with 61 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
atroposlib/envs/base.py
View file

@ -990,7 +990,12 @@ class BaseEnv(ABC):
if self.config.include_messages and group.get("messages") is None:
group["messages"] = [
self.tokenizer.decode(group["tokens"][i])
[
{
"role": "user",
"content": self.tokenizer.decode(group["tokens"][i]),
}
]
for i in range(len(group["tokens"]))
]