Skip to content

Commit 0a084b5

Browse files
committed
fix test due to vllm-project/vllm#17130
Signed-off-by: Linkun Chen <[email protected]>
1 parent 1af345e commit 0a084b5

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

python/ray/llm/tests/serve/mocks/mock_vllm_engine.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -584,7 +584,13 @@ async def _start_engine(self) -> EngineClient:
584584
return MockPDDisaggVLLMEngineClient(
585585
VllmConfig(
586586
model_config=ModelConfig(
587-
model=self.llm_config.model_loading_config.model_id
587+
model=self.llm_config.model_loading_config.model_id,
588+
task="auto",
589+
tokenizer=None,
590+
tokenizer_mode="auto",
591+
trust_remote_code=False,
592+
dtype="auto",
593+
seed=None,
588594
)
589595
)
590596
)

0 commit comments

Comments
 (0)