More tweaks to cli args

comfy/cli_args.py

@@ -124,15 +124,13 @@ parser.add_argument("--distributed-queue-connection-uri", type=str, default=None
 parser.add_argument(
     '--distributed-queue-worker',
     required=False,
-    type=bool,
-    default=True,
+    action="store_true",
     help='Workers will pull requests off the AMQP URL.'
 )
 parser.add_argument(
     '--distributed-queue-frontend',
     required=False,
-    type=bool,
-    default=True,
+    action="store_true",
     help='Frontends will start the web UI and connect to the provided AMQP URL to submit prompts.'
 )
 parser.add_argument("--distributed-queue-name", type=str, default="comfyui",

comfy/cli_args_types.py

@@ -132,8 +132,8 @@ class Configuration(dict):
         self.write_out_config_file: bool = False
         self.create_directories: bool = False
         self.distributed_queue_connection_uri: Optional[str] = None
-        self.distributed_queue_worker: bool = True
-        self.distributed_queue_frontend: bool = True
+        self.distributed_queue_worker: bool = False
+        self.distributed_queue_frontend: bool = False
         self.distributed_queue_name: str = "comfyui"
         for key, value in kwargs.items():
             self[key] = value

comfy/cmd/main.py

@@ -298,8 +298,9 @@ async def main():
     try:
         await run(server, address=args.listen, port=args.port, verbose=not args.dont_print_server,
                                     call_on_start=call_on_start)
-    except KeyboardInterrupt:
-        await q.close()
+    except asyncio.CancelledError:
+        if distributed:
+            await q.close()
         print("\nStopped server")
 
     cleanup_temp()