Optimise ComfyUI

fofr · Mar 19, 2024 · b2b7563 · b2b7563
1 parent 7363274
commit b2b7563
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/become-image-api.json b/become-image-api.json
@@ -67,7 +67,7 @@
   },
   "36": {
     "inputs": {
-      "provider": "CPU"
+      "provider": "CUDA"
     },
     "class_type": "InstantIDFaceAnalysis",
     "_meta": {
@@ -124,8 +124,8 @@
   },
   "49": {
     "inputs": {
-      "preprocessor": "Zoe-DepthMapPreprocessor",
-      "resolution": 1024,
+      "preprocessor": "MiDaS-DepthMapPreprocessor",
+      "resolution": 512,
       "image": [
         "67",
         0

diff --git a/helpers/comfyui.py b/helpers/comfyui.py
@@ -47,7 +47,7 @@ def start_server(self, output_directory, input_directory):
         print("Server running")
 
     def run_server(self, output_directory, input_directory):
-        command = f"python ./ComfyUI/main.py --output-directory {output_directory} --input-directory {input_directory} --disable-metadata"
+        command = f"python ./ComfyUI/main.py --output-directory {output_directory} --input-directory {input_directory} --disable-metadata --preview-method none --gpu-only"
         server_process = subprocess.Popen(command, shell=True)
         server_process.wait()