Skip to content

vllm.entrypoints.cli.benchmark.sweep

Classes:

BenchmarkSweepSubcommand

Bases: BenchmarkSubcommandBase

The sweep subcommand for vllm bench.

Source code in vllm/entrypoints/cli/benchmark/sweep.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
class BenchmarkSweepSubcommand(BenchmarkSubcommandBase):
    """The `sweep` subcommand for `vllm bench`."""

    name = "sweep"
    help = "Benchmark for a parameter sweep."

    @classmethod
    def add_cli_args(cls, parser: argparse.ArgumentParser) -> None:
        add_cli_args(parser)

    @staticmethod
    def cmd(args: argparse.Namespace) -> None:
        main(args)