From d407d823500b327d56a8d59c499ae4fb5e9166ab Mon Sep 17 00:00:00 2001
From: Tara Ding <38710454+windtara0619@users.noreply.github.com>
Date: Mon, 27 Apr 2026 14:20:32 -0700
Subject: [PATCH] Fix comments

---
 benchmarks/benchmark_comfyui_serving.py | 35 +++++++++----------------
 1 file changed, 13 insertions(+), 22 deletions(-)

diff --git a/benchmarks/benchmark_comfyui_serving.py b/benchmarks/benchmark_comfyui_serving.py
index 1855912da..2629833e2 100644
--- a/benchmarks/benchmark_comfyui_serving.py
+++ b/benchmarks/benchmark_comfyui_serving.py
@@ -1,36 +1,27 @@
 #!/usr/bin/env python3
 """
-Simple serving benchmark client for ComfyUI's HTTP API.
+ComfyUI model serving benchmark.
 
-This script is designed to:
-  - submit prompts to ComfyUI (/prompt or /bench/prompt),
-  - optionally shape request arrivals (fixed rate or Poisson),
-  - poll completion via /history/{prompt_id},
-  - report latency/throughput/error metrics.
+Submits prompts concurrently to a running ComfyUI server and reports
+latency/throughput metrics. Input images and prompt files are prepared
+automatically (and cached for reuse) before the benchmark starts.
 
-Usage — Wan 2.2 I2V benchmark
-==============================
+Supported models / tasks
+------------------------
+  wan22 / i2v   — Wan 2.2 Image-to-Video (LightX2V 4-step, 720×720, 81 frames)
 
-Images and prompt files are prepared automatically on first run and reused on
-subsequent runs. Just specify --model and --task:
-
-  # Minimal (synthetic images, default paths):
+Usage
+-----
   python3 benchmarks/benchmark_comfyui_serving.py \\
     --model wan22 --task i2v \\
-    --num-requests 50 --max-concurrency 4
+    --num-requests 50 --max-concurrency 4 \\
+    --host http://127.0.0.1:8188
 
-  # With model download (needs ComfyUI root):
+  # Also download model weights (run from ComfyUI root):
   python3 benchmarks/benchmark_comfyui_serving.py \\
     --model wan22 --task i2v \\
     --download-models --comfyui-base-dir /path/to/ComfyUI \\
-    --num-requests 50 --max-concurrency 4
-
-  # Custom paths:
-  python3 benchmarks/benchmark_comfyui_serving.py \\
-    --model wan22 --task i2v \\
-    --input-dir /home/ubuntu/ComfyUI/input \\
-    --prompts-dir /home/ubuntu/ComfyUI/benchmarks/prompts/wan22_i2v \\
-    --num-images 30 --num-requests 50 --max-concurrency 4 \\
+    --num-requests 50 --max-concurrency 4 \\
     --host http://127.0.0.1:8188
 """