From 61057bda6c4b8cda1117d74f100d0735645c0cfb Mon Sep 17 00:00:00 2001
From: "Wang, FangYuan" <39615225+At1a8@users.noreply.github.com>
Date: Sun, 26 Jul 2026 15:07:54 +0800
Subject: [PATCH] [BugFix] Prevent TBO crash when return_logprob is enabled
 (#32180)

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

diff --git a/python/sglang/srt/batch_overlap/two_batch_overlap.py b/python/sglang/srt/batch_overlap/two_batch_overlap.py
index 8352086bb..326d313df 100644
--- a/python/sglang/srt/batch_overlap/two_batch_overlap.py
+++ b/python/sglang/srt/batch_overlap/two_batch_overlap.py
@@ -796,6 +796,7 @@ class TboForwardBatchPreparer:
                 mm_inputs=None,
                 top_logprobs_nums=None,
                 token_ids_logprobs=None,
+                extend_input_logprob_token_ids_gpu=None,
                 next_token_logits_buffer=None,
                 return_hidden_states_before_norm=False,
                 # TBO children start unplanned — planned by the TBO-aware init