Skip to content

Commit a291439

Browse files
authored
Support logprobs in two-batch overlap (sgl-project#7709)
1 parent 54411f6 commit a291439

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/sglang/srt/two_batch_overlap.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -550,6 +550,8 @@ def filter_batch(
550550
top_p_normalized_logprobs=False,
551551
top_p=None,
552552
mm_inputs=None,
553+
top_logprobs_nums=None,
554+
token_ids_logprobs=None,
553555
)
554556
)
555557

0 commit comments

Comments
 (0)