Skip to content

Commit f05a13c

Browse files
authored
[TPU] Fix tpu model runner test (vllm-project#19995)
Signed-off-by: Chenyaaang <chenyangli@google.com>
1 parent faca98b commit f05a13c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/v1/tpu/worker/test_tpu_model_runner.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from vllm.attention.layer import Attention
77
from vllm.config import (CacheConfig, ModelConfig, SchedulerConfig, VllmConfig,
88
set_current_vllm_config)
9+
from vllm.pooling_params import PoolingParams
910
from vllm.sampling_params import SamplingParams
1011
from vllm.utils import GiB_bytes
1112
from vllm.v1.core.kv_cache_utils import (estimate_max_model_len,
@@ -71,6 +72,7 @@ def _schedule_new_request(*req_ids: str) -> SchedulerOutput:
7172
mm_hashes=[],
7273
mm_positions=[],
7374
sampling_params=SamplingParams(),
75+
pooling_params=PoolingParams(),
7476
block_ids=([0], ), # block_ids should be tuple[list[int]]
7577
num_computed_tokens=0,
7678
lora_request=None,

0 commit comments

Comments
 (0)