Skip to content

Commit 970340f

Browse files
hmelloramd-xiaoyu12
authored andcommitted
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Signed-off-by: Xiao Yu <xiao.yu@amd.com>
1 parent daa8a03 commit 970340f

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

vllm/config/compilation.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,8 @@ class CompilationConfig:
225225
# CudaGraph compilation
226226
cudagraph_mode: Optional[CUDAGraphMode] = None
227227
"""
228-
The mode of the cudagraph.
228+
The mode of the cudagraph:
229+
229230
- NONE, no cudagraph capture.
230231
- PIECEWISE. (v1 default)
231232
- FULL.
@@ -384,13 +385,10 @@ def __repr__(self) -> str:
384385
if pass_config_exclude:
385386
exclude["pass_config"] = pass_config_exclude
386387

387-
# The cast to string is necessary because Pydantic is mocked in docs
388-
# builds and sphinx-argparse doesn't know the return type of decode()
389-
return str(
390-
TypeAdapter(CompilationConfig).dump_json(
391-
self,
392-
exclude=exclude, # type: ignore[arg-type]
393-
exclude_unset=True).decode())
388+
return TypeAdapter(CompilationConfig).dump_json(
389+
self,
390+
exclude=exclude, # type: ignore[arg-type]
391+
exclude_unset=True).decode()
394392

395393
__str__ = __repr__
396394

0 commit comments

Comments
 (0)