We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
SpeculativeConfig
1 parent ad344ef commit 4fbd8bbCopy full SHA for 4fbd8bb
vllm/engine/arg_utils.py
@@ -826,6 +826,10 @@ def add_cli_args(parser: FlexibleArgumentParser) -> FlexibleArgumentParser:
826
title="VllmConfig",
827
description=VllmConfig.__doc__,
828
)
829
+ # We construct SpeculativeConfig using fields from other configs in
830
+ # create_engine_config. So we set the type to a JSON string here to
831
+ # delay the Pydantic validation that comes with SpeculativeConfig.
832
+ vllm_kwargs["speculative_config"]["type"] = optional_type(json.loads)
833
vllm_group.add_argument("--speculative-config",
834
**vllm_kwargs["speculative_config"])
835
vllm_group.add_argument("--kv-transfer-config",
0 commit comments