refactor: remove ModelWorkerBatch indirection (#25516)

This commit is contained in:
Liangsheng Yin
2026-05-17 18:05:39 -07:00
committed by GitHub
parent b3803164cb
commit 58ece60703
21 changed files with 430 additions and 559 deletions
+1 -2
View File
@@ -121,8 +121,7 @@ class TestForwardSplitPrefill(CustomTestCase):
batch.prepare_for_extend()
# Create forward batch
model_worker_batch = batch.get_model_worker_batch()
forward_batch = ForwardBatch.init_new(model_worker_batch, self.model_runner)
forward_batch = ForwardBatch.init_new(batch, self.model_runner)
return forward_batch