Refactor: Moving extend_logprob_start_len calculation out of prepare_for_extend (#16105)

2025-12-30 12:38:33 +08:00
parent 9e263c2162
commit 60f1ca6925
7 changed files with 25 additions and 36 deletions
@@ -68,6 +68,7 @@ class TestPrefillAdder(CustomTestCase):
        req.rid = str(rid)
        req.priority = priority
        req.extend_input_len = 0
+        req.extend_logprob_start_len = 0
        req.output_ids = [0] * output_len
        req.sampling_params = SimpleNamespace(max_new_tokens=max_new_tokens)
        req.time_stats = SimpleNamespace(wait_queue_entry_time=wait_time)