[MISC] Fix misleading batch_size_capture_list when cuda_graph_sizes < 4 (#25829)

Signed-off-by: billishyahao <[email protected]>
Co-authored-by: Luka Govedic <[email protected]>

Author: billishyahao

vllm/config/vllm.py

@@ -580,9 +580,12 @@ def _set_cudagraph_sizes(self):
             not self.model_config.enforce_eager:
             cuda_graph_sizes = self.scheduler_config.cuda_graph_sizes
             if len(cuda_graph_sizes) == 1:
-                batch_size_capture_list = [1, 2, 4] + [
-                    i for i in range(8, cuda_graph_sizes[0] + 1, 8)
-                ]
+                max_graph_size = cuda_graph_sizes[0]
+                assert max_graph_size >= 1, "Maximum cudagraph size should be" \
+                                            " greater than or equal to 1."
+                batch_size_capture_list = [
+                    i for i in [1, 2, 4] if i <= max_graph_size
+                ] + list(range(8, max_graph_size + 1, 8))
             elif len(cuda_graph_sizes) > 1:
                 batch_size_capture_list = sorted(cuda_graph_sizes)
             else: