From a0f7c2ca6f674d73ecff86f561b8d408aa2b303a Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 16 Aug 2024 22:39:07 +0300
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 700481e..3588e07 100644
--- a/nodes.py
+++ b/nodes.py
@@ -153,7 +153,7 @@ class InitFluxTraining:
         mm.soft_empty_cache()
 
         parser = setup_parser()
-        args = parser.parse_args()
+        args = parser.parse_known_args()
 
         if kwargs.get("cache_latents") == "memory":
             kwargs["cache_latents"] = True