From a291439a59d9b53e7700a4e9560ad7d6995c366a Mon Sep 17 00:00:00 2001
From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
Date: Sun, 6 Jul 2025 10:05:32 +0800
Subject: [PATCH] Support logprobs in two-batch overlap (#7709)

---
 python/sglang/srt/two_batch_overlap.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/sglang/srt/two_batch_overlap.py b/python/sglang/srt/two_batch_overlap.py
index 2d32b118a..6d2d73028 100644
--- a/python/sglang/srt/two_batch_overlap.py
+++ b/python/sglang/srt/two_batch_overlap.py
@@ -550,6 +550,8 @@ class TboForwardBatchPreparer:
                 top_p_normalized_logprobs=False,
                 top_p=None,
                 mm_inputs=None,
+                top_logprobs_nums=None,
+                token_ids_logprobs=None,
             )
         )