From ffc6a7fd38d93964033f50c6d6fd36ffed1b7f07 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Tue, 18 Feb 2025 19:46:57 -0800 Subject: [PATCH] Use spawn multiprocessing context to fix Linux ProcessPool issues --- comfy/distributed/process_pool_executor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/comfy/distributed/process_pool_executor.py b/comfy/distributed/process_pool_executor.py index 96f86ac77..b0b7853bb 100644 --- a/comfy/distributed/process_pool_executor.py +++ b/comfy/distributed/process_pool_executor.py @@ -21,7 +21,9 @@ class ProcessPoolExecutor(ProcessPool, Executor): max_tasks: int = 0, initializer: Callable = None, initargs: list | tuple = (), - context: multiprocessing.context.BaseContext = multiprocessing): + context: multiprocessing.context.BaseContext = None): + if context is None: + context = multiprocessing.get_context('spawn') super().__init__(max_workers=max_workers, max_tasks=max_tasks, initializer=initializer, initargs=initargs, context=context) def shutdown(self, wait=True, *, cancel_futures=False):