diff --git a/comfy/cmd/main.py b/comfy/cmd/main.py index 2348455e8..41695d6c0 100644 --- a/comfy/cmd/main.py +++ b/comfy/cmd/main.py @@ -257,9 +257,9 @@ async def main(): # in a distributed setting, the default prompt worker will not be able to send execution events via the websocket worker_thread_server = server if not distributed else ServerStub() - if "worker" in args.distributed_queue_roles: - logging.warning( - f"Distributed workers started in the default thread loop cannot notify clients of progress updates. Instead of comfyui or main.py, use comfyui-worker.") + if not distributed or "worker" in args.distributed_queue_roles: + if distributed: + logging.warning(f"Distributed workers started in the default thread loop cannot notify clients of progress updates. Instead of comfyui or main.py, use comfyui-worker.") threading.Thread(target=prompt_worker, daemon=True, args=(q, worker_thread_server,)).start() # server has been imported and things should be looking good