mirror of
https://github.com/vllm-project/vllm.git
synced 2026-08-23 22:20:15 +00:00
[Core] Fix redundant None append in StepPool.forward for chunked prefill (#41049)
Signed-off-by: Anthony Su <[email protected]>
This commit is contained in:
@@ -111,7 +111,7 @@ class StepPool(AllPool):
|
||||
if step_tag_id is not None:
|
||||
data = data[token_id == step_tag_id]
|
||||
|
||||
pooled_data.append(data)
|
||||
pooled_data.append(data)
|
||||
|
||||
return pooled_data
|
||||
|
||||
|
||||
Reference in New Issue
Block a user