[SRT] minor: reuse req input id array for unpadded ids (#26232)

2026-05-26 08:58:32 +08:00
parent 8805f4cf16
commit 8f2a4e70f8
1 changed files with 5 additions and 3 deletions
@@ -690,10 +690,12 @@ class Req(ReqDllmMixin):
    ):
        # Input and output info
        self.rid = rid
-        self.origin_input_ids_unpadded = array(
-            "q", origin_input_ids_unpadded or origin_input_ids
-        )  # Before image padding
        self.origin_input_ids = array("q", origin_input_ids)
+        self.origin_input_ids_unpadded = (
+            array("q", origin_input_ids_unpadded)
+            if origin_input_ids_unpadded
+            else self.origin_input_ids
+        )  # Before image padding
        # Each decode stage's output ids
        self.output_ids = array("q")
        # fill_ids = origin_input_ids + output_ids. Updated if chunked.