More tweaks to cli args

This commit is contained in:
doctorpangloss 2024-02-09 01:40:27 -08:00
parent a3f9d007d4
commit f195230e2a
3 changed files with 7 additions and 8 deletions

View File

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

View File

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

View File

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