From c9676345e7303270efb2d03f449462f40f584def Mon Sep 17 00:00:00 2001 From: Robert Wojciechowski Date: Tue, 24 Feb 2026 01:39:56 +0000 Subject: [PATCH] fix: use centralized tile state init in distributed startup --- distributed.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/distributed.py b/distributed.py index 9c02da5..0f9eb64 100644 --- a/distributed.py +++ b/distributed.py @@ -12,6 +12,7 @@ from .utils.logging import debug_log from .utils.network import cleanup_client_session from .workers import get_worker_manager from .workers.startup import delayed_auto_launch, register_async_signals, sync_cleanup +from .upscale.job_store import ensure_tile_jobs_initialized from .nodes import ( NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS, @@ -41,10 +42,7 @@ atexit.register(lambda: None) # placeholder; real cleanup in sync_cleanup # Initialize distributed job state on prompt_server prompt_server = server.PromptServer.instance ensure_distributed_state(prompt_server) - -if not hasattr(prompt_server, 'distributed_pending_tile_jobs'): - prompt_server.distributed_pending_tile_jobs = {} - prompt_server.distributed_tile_jobs_lock = asyncio.Lock() +ensure_tile_jobs_initialized() # Worker startup if not os.environ.get('COMFYUI_IS_WORKER'):