Fix edge case for truncated (#1747)

17536e7e · Byron Hsu · GitHub · 1f26e8b8 · 17536e7e
Unverified Commit 17536e7e authored Oct 22, 2024 by Byron Hsu Committed by GitHub Oct 23, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/sglang/srt/managers/scheduler.py python/sglang/srt/managers/scheduler.py +1 -1

No files found.
--- a/python/sglang/srt/managers/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -416,7 +416,7 @@ class Scheduler:
                )

        # Truncate prompts that are too long
-        if len(req.origin_input_ids) >= self.max_req_input_len:
+        if len(req.origin_input_ids) > self.max_req_input_len:
            logger.warning(
                "Request length is longer than the KV cache pool size or "
                "the max context length. Truncated!!!"