[BugFix] Fix draft model specified config file (#17815)
This commit is contained in:
@@ -241,6 +241,11 @@ class ModelConfig:
|
|||||||
if is_draft_model
|
if is_draft_model
|
||||||
else server_args.quantization
|
else server_args.quantization
|
||||||
)
|
)
|
||||||
|
override_config_file = (
|
||||||
|
server_args.decrypted_draft_config_file
|
||||||
|
if is_draft_model
|
||||||
|
else server_args.decrypted_config_file
|
||||||
|
)
|
||||||
return ModelConfig(
|
return ModelConfig(
|
||||||
model_path=model_path or server_args.model_path,
|
model_path=model_path or server_args.model_path,
|
||||||
trust_remote_code=server_args.trust_remote_code,
|
trust_remote_code=server_args.trust_remote_code,
|
||||||
@@ -254,7 +259,7 @@ class ModelConfig:
|
|||||||
model_impl=server_args.model_impl,
|
model_impl=server_args.model_impl,
|
||||||
sampling_defaults=server_args.sampling_defaults,
|
sampling_defaults=server_args.sampling_defaults,
|
||||||
quantize_and_serve=server_args.quantize_and_serve,
|
quantize_and_serve=server_args.quantize_and_serve,
|
||||||
override_config_file=server_args.decrypted_config_file,
|
override_config_file=override_config_file,
|
||||||
is_multi_layer_eagle=server_args.enable_multi_layer_eagle,
|
is_multi_layer_eagle=server_args.enable_multi_layer_eagle,
|
||||||
language_only=server_args.language_only,
|
language_only=server_args.language_only,
|
||||||
encoder_only=server_args.encoder_only,
|
encoder_only=server_args.encoder_only,
|
||||||
|
|||||||
Reference in New Issue
Block a user