Unverified Commit d57e3d65 authored by Younes B's avatar Younes B Committed by GitHub
Browse files

fix: fix vllm issue with DP>1 (#3025)

parent 3f792954
...@@ -216,9 +216,17 @@ class VLLM(TemplateLM): ...@@ -216,9 +216,17 @@ class VLLM(TemplateLM):
} }
if parse_version(version("vllm")) >= parse_version("0.9.0"): if parse_version(version("vllm")) >= parse_version("0.9.0"):
if self.data_parallel_size <= 1:
kwargs_resolve_hf_chat_template["model_config"] = ( kwargs_resolve_hf_chat_template["model_config"] = (
self.model.llm_engine.model_config self.model.llm_engine.model_config
) )
else:
from vllm.engine.arg_utils import EngineArgs
engine_args = EngineArgs(**self.model_args)
model_config = engine_args.create_model_config()
kwargs_resolve_hf_chat_template["model_config"] = model_config
# https://github.com/vllm-project/vllm/pull/18259 # https://github.com/vllm-project/vllm/pull/18259
if ( if (
......
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