mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-14 02:57:24 +08:00
add tqdm to the benchmark
This commit is contained in:
parent
ba978bc0e2
commit
c39f7ea76c
@ -51,6 +51,7 @@ from pathlib import Path
|
|||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
import aiohttp
|
import aiohttp
|
||||||
|
from tqdm import tqdm
|
||||||
|
|
||||||
|
|
||||||
# ──────────────────────────────────────────────────────────────────────────────
|
# ──────────────────────────────────────────────────────────────────────────────
|
||||||
@ -669,7 +670,14 @@ async def async_main(args: argparse.Namespace) -> None:
|
|||||||
)
|
)
|
||||||
for i in range(args.num_requests)
|
for i in range(args.num_requests)
|
||||||
]
|
]
|
||||||
results = await asyncio.gather(*tasks)
|
results = []
|
||||||
|
with tqdm(total=args.num_requests, unit="req", desc="benchmark") as pbar:
|
||||||
|
for coro in asyncio.as_completed(tasks):
|
||||||
|
result = await coro
|
||||||
|
results.append(result)
|
||||||
|
pbar.update(1)
|
||||||
|
if result.ok:
|
||||||
|
pbar.set_postfix(succeeded=sum(r.ok for r in results))
|
||||||
wall_s = time.perf_counter() - started
|
wall_s = time.perf_counter() - started
|
||||||
|
|
||||||
print_summary(results, wall_s)
|
print_summary(results, wall_s)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user