Fix customized_info offset truncation (#21262)

2026-03-26 10:57:51 +08:00
parent 1100b9865c
commit 53c1d8e963
1 changed files with 3 additions and 1 deletions
@@ -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()