File tree Expand file tree Collapse file tree 2 files changed +0
-3
lines changed Expand file tree Collapse file tree 2 files changed +0
-3
lines changed Original file line number Diff line number Diff 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,
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments