Add 10 min timeout for websocket

Comfy-Org · Jul 18, 2024 · f5818ad · f5818ad
1 parent 50a0609
commit f5818ad
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/action.py b/action.py
@@ -159,7 +159,16 @@ def main(args):
         start_time = int(datetime.datetime.now().timestamp())
         try:
             result = subprocess.run(
-                ["comfy", "--skip-prompt", "--no-enable-telemetry", "run", "--workflow", file_path],
+                [
+                    "comfy",
+                    "--skip-prompt",
+                    "--no-enable-telemetry",
+                    "run",
+                    "--workflow",
+                    file_path,
+                    "--timeout",
+                    600
+                ],
                 check=True,
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,