diff --git a/vllm/config/compilation.py b/vllm/config/compilation.py index 825de7d261918..df08fa20e0cfa 100644 --- a/vllm/config/compilation.py +++ b/vllm/config/compilation.py @@ -73,6 +73,9 @@ class CUDAGraphMode(enum.Enum): CUDAGraphMode.NONE, CUDAGraphMode.PIECEWISE, CUDAGraphMode.FULL ] + def __str__(self) -> str: + return self.name + @config @dataclass @@ -417,10 +420,11 @@ class CompilationConfig: if pass_config_exclude: exclude["pass_config"] = pass_config_exclude - return TypeAdapter(CompilationConfig).dump_json( - self, - exclude=exclude, # type: ignore[arg-type] - exclude_unset=True).decode() + config = TypeAdapter(CompilationConfig).dump_python(self, + exclude=exclude, + exclude_unset=True) + + return str(config) __str__ = __repr__