Skip to content

GitLab

Unverified Commit b2f77457 authored by cong-meta's avatar cong-meta Committed by GitHub
Browse files

Add data_parallel_size to VllmConfig string representation (#24298) 


Co-authored-by: default avatarCong Chen <congc@meta.com>
parent 82dfb12e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
vllm/config/__init__.py
View file @ b2f77457
...@@ -3790,6 +3790,7 @@ class VllmConfig: ...@@ -3790,6 +3790,7 @@ class VllmConfig:
f"load_format={self.load_config.load_format}, " f"load_format={self.load_config.load_format}, "
f"tensor_parallel_size={self.parallel_config.tensor_parallel_size}, " # noqa f"tensor_parallel_size={self.parallel_config.tensor_parallel_size}, " # noqa
f"pipeline_parallel_size={self.parallel_config.pipeline_parallel_size}, " # noqa f"pipeline_parallel_size={self.parallel_config.pipeline_parallel_size}, " # noqa
f"data_parallel_size={self.parallel_config.data_parallel_size}, " # noqa
f"disable_custom_all_reduce={self.parallel_config.disable_custom_all_reduce}, " # noqa f"disable_custom_all_reduce={self.parallel_config.disable_custom_all_reduce}, " # noqa
f"quantization={self.model_config.quantization}, " f"quantization={self.model_config.quantization}, "
f"enforce_eager={self.model_config.enforce_eager}, " f"enforce_eager={self.model_config.enforce_eager}, "
......
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