support vlm benchmark profile (#5905)

This commit is contained in:
Yi Zhang
2025-04-30 14:48:27 +08:00
committed by GitHub
parent 8fefdd32c7
commit d50e36a79d
2 changed files with 70 additions and 3 deletions

View File

@@ -33,6 +33,8 @@ class EvalArgs:
prompt_format_file: str = "prompt_format.yaml"
dataset_path: str = "MMMU/MMMU"
extra_request_body: Optional[str] = None
profile: bool = False
profile_number: int = 5
@staticmethod
def add_cli_args(parser: argparse.ArgumentParser):
@@ -65,6 +67,12 @@ class EvalArgs:
help="Append given JSON object to the request payload. You can use this to specify"
"additional generate params like sampling params.",
)
parser.add_argument(
"--profile", action="store_true", help="enable mmmu profile"
)
parser.add_argument(
"--profile-number", type=int, default=EvalArgs.profile_number
)
@classmethod
def from_cli_args(cls, args: argparse.Namespace):