From 08411a1d65a6e42db5611984089c71d8c654c83e Mon Sep 17 00:00:00 2001
From: Tara Ding <38710454+windtara0619@users.noreply.github.com>
Date: Mon, 27 Apr 2026 14:28:46 -0700
Subject: [PATCH] Fix input dir

---
 benchmarks/benchmark_comfyui_serving.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/benchmarks/benchmark_comfyui_serving.py b/benchmarks/benchmark_comfyui_serving.py
index 2236230bd..91c81bcf3 100644
--- a/benchmarks/benchmark_comfyui_serving.py
+++ b/benchmarks/benchmark_comfyui_serving.py
@@ -598,12 +598,6 @@ def parse_args() -> argparse.Namespace:
         required=True,
         help=f"Task type. Choices: {_VALID_TASKS}.",
     )
-    p.add_argument(
-        "--input-dir",
-        type=Path,
-        default=Path("input"),
-        help="ComfyUI input image directory (default: input/). LoadImage resolves files from this folder.",
-    )
     p.add_argument(
         "--prompts-dir",
         type=Path,
@@ -652,7 +646,7 @@ async def async_main(args: argparse.Namespace) -> None:
         model=args.model,
         task=args.task,
         output_dir=prompts_dir,
-        input_dir=args.input_dir,
+        input_dir=Path("input"),
         num_images=args.num_images,
         download_model_weights=args.download_models,
         comfyui_base_dir=args.comfyui_base_dir,