Add throughput benchmarking script (#133)

This commit is contained in:
Woosuk Kwon
2023-05-28 03:20:05 -07:00
committed by GitHub
parent 337871c6fd
commit 211318d44a
12 changed files with 145 additions and 257 deletions

View File

@@ -19,9 +19,8 @@ def main(args: argparse.Namespace):
SamplingParams(n=3, best_of=3, use_beam_search=True, temperature=0.0)),
]
request_id = 0
# Run the server.
request_id = 0
while True:
# To test iteration-level scheduling, we add one request at each step.
if test_prompts: