forked from Karylab-cklius/vllm
[Core] Fix redundant None append in StepPool.forward for chunked prefill (#41049)
Signed-off-by: Anthony Su <xsuanthony@gmail.com>
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