[Bugfix] Add rids to the batch filtering for two batch overlap (#19418)

This commit is contained in:
Shangming Cai
2026-02-26 22:57:25 +08:00
committed by GitHub
parent 97f1fa5e6b
commit e55e65535e
2 changed files with 2 additions and 1 deletions

View File

@@ -653,6 +653,7 @@ class TboForwardBatchPreparer:
"extend_seq_lens_cpu",
"extend_logprob_start_lens_cpu",
"lora_ids",
"rids",
]:
old_value = getattr(batch, key)
if old_value is None: