From aada34288f7d70f08aa5424a20441369739e9a54 Mon Sep 17 00:00:00 2001
From: Alexander Piskun <13381981+bigcat88@users.noreply.github.com>
Date: Fri, 28 Feb 2025 08:46:42 +0200
Subject: [PATCH] allow custom command-line args

Signed-off-by: bigcat88 <bigcat88@icloud.com>
---
 comfy/cli_args.py | 4 ++--
 main.py           | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/comfy/cli_args.py b/comfy/cli_args.py
index a906ff1c0..7d54d8d15 100644
--- a/comfy/cli_args.py
+++ b/comfy/cli_args.py
@@ -182,9 +182,9 @@ parser.add_argument("--user-directory", type=is_valid_directory, default=None, h
 parser.add_argument("--enable-compress-response-body", action="store_true", help="Enable compressing response body.")
 
 if comfy.options.args_parsing:
-    args = parser.parse_args()
+    args, unknown_args = parser.parse_known_args()
 else:
-    args = parser.parse_args([])
+    args, unknown_args = parser.parse_known_args([])
 
 if args.windows_standalone_build:
     args.auto_launch = True
diff --git a/main.py b/main.py
index f6510c90a..9fb9f8931 100644
--- a/main.py
+++ b/main.py
@@ -5,7 +5,7 @@ import os
 import importlib.util
 import folder_paths
 import time
-from comfy.cli_args import args
+from comfy.cli_args import args, unknown_args
 from app.logger import setup_logger
 import itertools
 import utils.extra_config
@@ -18,6 +18,8 @@ if __name__ == "__main__":
 
 
 setup_logger(log_level=args.verbose, use_stdout=args.log_stdout)
+if unknown_args:
+    logging.info("Found arguments not defined in the ComfyUI core: %s", unknown_args)
 
 def apply_custom_paths():
     # extra model paths