Skip to content

GitLab

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

[Bugfix] Fix `task` still being passed in tests/benchmarks (#30476) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent a5f9fb59
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
benchmarks/benchmark_ngram_proposer.py
View file @ d917747c
...@@ -32,7 +32,6 @@ def benchmark_propose(args): ...@@ -32,7 +32,6 @@ def benchmark_propose(args):
model_config = ModelConfig( model_config = ModelConfig(
model="facebook/opt-125m", model="facebook/opt-125m",
task="generate",
max_model_len=args.num_token + args.num_spec_token, max_model_len=args.num_token + args.num_spec_token,
tokenizer="facebook/opt-125m", tokenizer="facebook/opt-125m",
tokenizer_mode="auto", tokenizer_mode="auto",
......
tests/models/language/pooling/test_mm_classifier_conversion.py
View file @ d917747c
...@@ -17,7 +17,6 @@ def test_idefics_multimodal( ...@@ -17,7 +17,6 @@ def test_idefics_multimodal(
with vllm_runner( with vllm_runner(
model_name="HuggingFaceM4/Idefics3-8B-Llama3", model_name="HuggingFaceM4/Idefics3-8B-Llama3",
runner="pooling", runner="pooling",
task="classify",
convert="classify", convert="classify",
load_format="dummy", load_format="dummy",
max_model_len=512, max_model_len=512,
...@@ -86,7 +85,6 @@ def test_gemma_multimodal( ...@@ -86,7 +85,6 @@ def test_gemma_multimodal(
with vllm_runner( with vllm_runner(
model_name="google/gemma-3-4b-it", model_name="google/gemma-3-4b-it",
runner="pooling", runner="pooling",
task="classify",
convert="classify", convert="classify",
load_format="auto", load_format="auto",
hf_overrides=update_config, hf_overrides=update_config,
......
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