From 722f25c9054336229edb8114ad2a137b1ee093b9 Mon Sep 17 00:00:00 2001
From: ZHANG Hao <zhanghao@4paradigm.com>
Date: Wed, 3 Sep 2025 15:53:21 +0800
Subject: [PATCH] ignore unknow args

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 5e77c0f..cdffc7e 100644
--- a/main.py
+++ b/main.py
@@ -97,7 +97,7 @@ def main():
     parser.add_argument("--outdir", required=True, help="图片输出目录")
     parser.add_argument("--device", default="cuda", choices=["cuda", "cpu"], help="推理设备")
     parser.add_argument("--dtype", default="fp16", choices=["fp16", "fp32"], help="推理精度")
-    args = parser.parse_args()
+    args, _ = p.parse_known_args()
 
     model_path = args.model
     json_path = Path(args.json)