Skip to content

Commit bf0ae2c

Browse files
committed
fix version
Signed-off-by: wangxiyuan <[email protected]>
1 parent 5bb6017 commit bf0ae2c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/worker/model_runner_v1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1816,7 +1816,7 @@ def _pool(
18161816
device=hidden_states.device)
18171817
seq_lens_cpu = self.seq_lens_cpu[:self.input_batch.num_reqs]
18181818

1819-
if vllm_version_is("0.12.0"):
1819+
if vllm_version_is("0.10.2"):
18201820
# Pooling models D2H & synchronize occurs in pooler.py:build_output
18211821
raw_pooler_output = self.model.pooler(
18221822
hidden_states=hidden_states, pooling_metadata=pooling_metadata)
@@ -1835,7 +1835,7 @@ def _pool(
18351835
pooler_output: list[Optional[torch.Tensor]] = []
18361836
for raw_output, seq_len, prompt_len in zip(
18371837
raw_pooler_output, seq_lens_cpu, pooling_metadata.prompt_lens):
1838-
if vllm_version_is("0.12.0"):
1838+
if vllm_version_is("0.10.2"):
18391839
output = raw_output.data if seq_len == prompt_len else None
18401840
else:
18411841
output = raw_output if seq_len == prompt_len else None

0 commit comments

Comments
 (0)