Unverified Commit 51c2e1fc authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[CI/Build] Split up models tests (#10069)


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent b09895a6
......@@ -241,11 +241,11 @@ def init_vllm_registered_model(
based on the arguments passed to the outer vLLM model.
"""
model_class, _ = ModelRegistry.resolve_model_cls(hf_config.architectures)
import copy
copied_config = copy.deepcopy(vllm_config)
copied_config.model_config.hf_config = hf_config
return model_class(vllm_config=copied_config, prefix=prefix)
return model_class(
vllm_config=vllm_config.with_hf_config(hf_config),
prefix=prefix,
)
@overload
......
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