handle args

lllyasviel · Aug 30, 2023 · f3b96ef · f3b96ef
1 parent 0425ea4
commit f3b96ef
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git 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
@@ -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()