Skip to content

GitLab

Unverified Commit bfb240cc authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[CI/Build] Fix Prithvi plugin test (#27393) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent e255d929
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/plugins_tests/test_io_processor_plugins.py
View file @ bfb240cc
...@@ -100,6 +100,7 @@ def test_prithvi_mae_plugin_offline(vllm_runner, model_name: str): ...@@ -100,6 +100,7 @@ def test_prithvi_mae_plugin_offline(vllm_runner, model_name: str):
model_name, model_name,
runner="pooling", runner="pooling",
skip_tokenizer_init=True, skip_tokenizer_init=True,
enable_mm_embeds=True,
trust_remote_code=True, trust_remote_code=True,
enforce_eager=True, enforce_eager=True,
# Limit the maximum number of parallel requests # Limit the maximum number of parallel requests
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment