Skip to content

Commit e39dc46

Browse files
authored
[CI] Pooling models mteb test disable enforce_eager (#26408)
Signed-off-by: wang.yuqi <[email protected]>
1 parent 10c75b5 commit e39dc46

File tree

2 files changed

+0
-3
lines changed

2 files changed

+0
-3
lines changed

tests/models/language/generation_ppl_test/ppl_utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,6 @@ def wikitext_ppl_test(
5151
gpu_memory_utilization=0.7,
5252
max_model_len=max_length,
5353
max_num_seqs=1,
54-
enforce_eager=True,
5554
**vllm_extra_kwargs,
5655
) as vllm_model:
5756
# Use max_num_seqs=1 to avoid OOM,

tests/models/language/pooling_mteb_test/mteb_utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,6 @@ def mteb_test_embed_models(
192192
model_info.name,
193193
runner="pooling",
194194
max_model_len=None,
195-
enforce_eager=True,
196195
**vllm_extra_kwargs,
197196
) as vllm_model:
198197
model_config = vllm_model.llm.llm_engine.model_config
@@ -349,7 +348,6 @@ def mteb_test_rerank_models(
349348
runner="pooling",
350349
max_model_len=None,
351350
max_num_seqs=8,
352-
enforce_eager=True,
353351
**vllm_extra_kwargs,
354352
) as vllm_model:
355353
model_config = vllm_model.llm.llm_engine.model_config

0 commit comments

Comments
 (0)