ignore unknow args

2025-09-03 15:53:21 +08:00
parent 1b590aad75
commit 722f25c905
1 changed files with 1 additions and 1 deletions
--- 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)