Skip to content

Commit d57e3d6

Browse files
fix: fix vllm issue with DP>1 (#3025)
1 parent 3f79295 commit d57e3d6

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

lm_eval/models/vllm_causallms.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -216,9 +216,17 @@ def __init__(
216216
}
217217

218218
if parse_version(version("vllm")) >= parse_version("0.9.0"):
219-
kwargs_resolve_hf_chat_template["model_config"] = (
220-
self.model.llm_engine.model_config
221-
)
219+
if self.data_parallel_size <= 1:
220+
kwargs_resolve_hf_chat_template["model_config"] = (
221+
self.model.llm_engine.model_config
222+
)
223+
else:
224+
from vllm.engine.arg_utils import EngineArgs
225+
226+
engine_args = EngineArgs(**self.model_args)
227+
model_config = engine_args.create_model_config()
228+
229+
kwargs_resolve_hf_chat_template["model_config"] = model_config
222230

223231
# https://github.com/vllm-project/vllm/pull/18259
224232
if (

0 commit comments

Comments
 (0)