diff --git a/vllm/model_executor/layers/pooler/tokwise/methods.py b/vllm/model_executor/layers/pooler/tokwise/methods.py index 9ee6e8527c9..523e8e4c9a9 100644 --- a/vllm/model_executor/layers/pooler/tokwise/methods.py +++ b/vllm/model_executor/layers/pooler/tokwise/methods.py @@ -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