From e39dc46f8fe61803032a5f51ba76f8fa03ba0b41 Mon Sep 17 00:00:00 2001 From: "wang.yuqi" Date: Wed, 8 Oct 2025 20:15:36 +0800 Subject: [PATCH] [CI] Pooling models mteb test disable enforce_eager (#26408) Signed-off-by: wang.yuqi --- tests/models/language/generation_ppl_test/ppl_utils.py | 1 - tests/models/language/pooling_mteb_test/mteb_utils.py | 2 -- 2 files changed, 3 deletions(-) diff --git a/tests/models/language/generation_ppl_test/ppl_utils.py b/tests/models/language/generation_ppl_test/ppl_utils.py index dcef365e99e7c..43f6066b1c85e 100644 --- a/tests/models/language/generation_ppl_test/ppl_utils.py +++ b/tests/models/language/generation_ppl_test/ppl_utils.py @@ -51,7 +51,6 @@ def wikitext_ppl_test( gpu_memory_utilization=0.7, max_model_len=max_length, max_num_seqs=1, - enforce_eager=True, **vllm_extra_kwargs, ) as vllm_model: # Use max_num_seqs=1 to avoid OOM, diff --git a/tests/models/language/pooling_mteb_test/mteb_utils.py b/tests/models/language/pooling_mteb_test/mteb_utils.py index a4a7f1b48d3d3..d96dc90416855 100644 --- a/tests/models/language/pooling_mteb_test/mteb_utils.py +++ b/tests/models/language/pooling_mteb_test/mteb_utils.py @@ -192,7 +192,6 @@ def mteb_test_embed_models( model_info.name, runner="pooling", max_model_len=None, - enforce_eager=True, **vllm_extra_kwargs, ) as vllm_model: model_config = vllm_model.llm.llm_engine.model_config @@ -349,7 +348,6 @@ def mteb_test_rerank_models( runner="pooling", max_model_len=None, max_num_seqs=8, - enforce_eager=True, **vllm_extra_kwargs, ) as vllm_model: model_config = vllm_model.llm.llm_engine.model_config