Skip to content

GitLab

Unverified Commit 9b1769dd authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Bugfix] Fix lint error (#17547) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent 61c299f8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
vllm/config.py
View file @ 9b1769dd
...@@ -4250,8 +4250,7 @@ class VllmConfig: ...@@ -4250,8 +4250,7 @@ class VllmConfig:
elif len(cuda_graph_sizes) > 1: elif len(cuda_graph_sizes) > 1:
batch_size_capture_list = sorted(cuda_graph_sizes) batch_size_capture_list = sorted(cuda_graph_sizes)
else: else:
raise TypeError( raise TypeError(f"Invalid value for {cuda_graph_sizes=}.")
f"Invalid value for {cuda_graph_sizes=}.")
if self.parallel_config.tensor_parallel_size > 1 and \ if self.parallel_config.tensor_parallel_size > 1 and \
self.compilation_config.pass_config.enable_sequence_parallelism: self.compilation_config.pass_config.enable_sequence_parallelism:
batch_size_capture_list = \ batch_size_capture_list = \
......
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