diff --git a/python/sglang/srt/managers/scheduler_output_processor_mixin.py b/python/sglang/srt/managers/scheduler_output_processor_mixin.py index a5b4fb9f3..4c759a718 100644 --- a/python/sglang/srt/managers/scheduler_output_processor_mixin.py +++ b/python/sglang/srt/managers/scheduler_output_processor_mixin.py @@ -1109,7 +1109,9 @@ class SchedulerOutputProcessorMixin: for k, v in req.customized_info.items(): if k not in customized_info: customized_info[k] = [] - customized_info[k].append(v[send_token_offset:]) + customized_info[k].append( + v[send_token_offset : len(output_ids_)] + ) if ( req.finished()