tokenizer bug

This commit is contained in:
Jai Suphavadeeprasit 2026-03-11 19:35:12 -04:00
parent b457a678ce
commit 2f371e03fc

View file

@ -907,7 +907,7 @@ class BaseEnv(ABC):
"ensure your trainer handles this appropriately."
)
elif abort_on_any_max_length_exceeded and any(
[len(x) >= self.max_token_len for x in group["tokens"]]
[len(x) > self.max_token_len for x in group["tokens"]]
):
logger.warning("Token length is too long in a group, skipping...")
continue