Add accuracy test to CI: MMLU (#882)

This commit is contained in:
Ying Sheng
2024-08-01 21:20:17 -07:00
committed by GitHub
parent 76e59088d8
commit ae7ee01a8e
24 changed files with 831 additions and 136 deletions

View File

@@ -21,7 +21,7 @@ import sys
import time
import traceback
import warnings
from argparse import ArgumentParser as FlexibleArgumentParser
from argparse import ArgumentParser
from dataclasses import dataclass, field
from datetime import datetime
from typing import AsyncGenerator, List, Optional, Tuple, Union
@@ -868,14 +868,12 @@ def set_ulimit(target_soft_limit=65535):
if __name__ == "__main__":
parser = FlexibleArgumentParser(
description="Benchmark the online serving throughput."
)
parser = ArgumentParser(description="Benchmark the online serving throughput.")
parser.add_argument(
"--backend",
type=str,
required=True,
choices=list(ASYNC_REQUEST_FUNCS.keys()),
default="sglang",
help="Must specify a backend, depending on the LLM Inference Engine.",
)
parser.add_argument(