mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-24 05:10:18 +08:00
More tweaks to cli args
This commit is contained in:
parent
a3f9d007d4
commit
f195230e2a
@ -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",
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
@ -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()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user