From f3b96ef9f2cacc6df3c8b346ee8ac30e5d8901a7 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Wed, 30 Aug 2023 06:46:28 -0700
Subject: [PATCH] handle args

---
 fooocus_version.py | 2 +-
 launch.py          | 9 +++++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/fooocus_version.py b/fooocus_version.py
index 1a93c00f..0ad26420 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1 +1 @@
-version = '1.0.42'
+version = '1.0.43'
diff --git a/launch.py b/launch.py
index f6be2dc1..92131d3e 100644
--- a/launch.py
+++ b/launch.py
@@ -72,15 +72,20 @@ def download_models():
     return
 
 
-def cuda_malloc():
+def clear_comfy_args():
     argv = sys.argv
     sys.argv = [sys.argv[0]]
-    import cuda_malloc
+    import comfy.cli_args
     sys.argv = argv
 
 
+def cuda_malloc():
+    import cuda_malloc
+
+
 prepare_environment()
 
+clear_comfy_args()
 # cuda_malloc()
 
 download_models()