Benchmark Updates (#382)

2024-04-24 02:23:01 +08:00
parent 150d7020ed
commit da19434c2f
18 changed files with 72 additions and 17 deletions
--- a/benchmark/multi_document_qa/bench_sglang.py
+++ b/benchmark/multi_document_qa/bench_sglang.py
@@ -43,7 +43,7 @@ def main(args):
    # Run requests
    tic = time.time()
    states = multi_document_qa.run_batch(
-        arguments, temperature=0, num_threads=args.parallel)
+        arguments, temperature=0, num_threads=args.parallel, progress_bar=True)
    latency = time.time() - tic

    # Compute accuracy