Skip to content

Commit 63424f4

Browse files
xgwangxgwclefourrier
authored
fix context size check in sglang model (#787)
Co-authored-by: xgw <[email protected]> Co-authored-by: Clémentine Fourrier <[email protected]>
1 parent bab9f6d commit 63424f4

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/lighteval/models/sglang/sglang_model.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,6 +274,11 @@ def _greedy_until(
274274
f"{context_size + max_new_tokens=} which is greater than {self.max_length=}. Truncating context to {self.max_length - max_new_tokens} tokens."
275275
)
276276
context_size = self.max_length - max_new_tokens
277+
if context_size < 0:
278+
logger.critical(
279+
f"{context_size=} is less than 0, either reduce the max_new_tokens or increase model max length."
280+
)
281+
raise ValueError("Context size is less than 0.")
277282
inputs = [input[-context_size:] for input in inputs]
278283
else:
279284
if context_size > self.max_length:

0 commit comments

Comments
 (0)