biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix nits from #20059 (#23548)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-08-26 11:27:20 +01:00
committed by GitHub
parent 50fede6634
commit b00e69f8ca
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
vllm/config/compilation.py
View File

@@ -225,7 +225,8 @@ class CompilationConfig:
# CudaGraph compilation # CudaGraph compilation
cudagraph_mode: Optional[CUDAGraphMode] = None cudagraph_mode: Optional[CUDAGraphMode] = None
""" """
The mode of the cudagraph. The mode of the cudagraph:
- NONE, no cudagraph capture. - NONE, no cudagraph capture.
- PIECEWISE. (v1 default) - PIECEWISE. (v1 default)
- FULL. - FULL.
@@ -384,13 +385,10 @@ class CompilationConfig:
if pass_config_exclude: if pass_config_exclude:
exclude["pass_config"] = pass_config_exclude exclude["pass_config"] = pass_config_exclude
# The cast to string is necessary because Pydantic is mocked in docs return TypeAdapter(CompilationConfig).dump_json(
# builds and sphinx-argparse doesn't know the return type of decode() self,
return str( exclude=exclude, # type: ignore[arg-type]
TypeAdapter(CompilationConfig).dump_json( exclude_unset=True).decode()
self,
exclude=exclude, # type: ignore[arg-type]
exclude_unset=True).decode())
__str__ = __repr__ __str__ = __repr__