wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 05:10:18 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Use spawn multiprocessing context to fix Linux ProcessPool issues

Browse Source
This commit is contained in:
Benjamin Berman 2025-02-18 19:46:57 -08:00
parent e65faca817
commit ffc6a7fd38
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/distributed/process_pool_executor.py
View File

@ -21,7 +21,9 @@ class ProcessPoolExecutor(ProcessPool, Executor):
max_tasks: int = 0, max_tasks: int = 0,
initializer: Callable = None, initializer: Callable = None,
initargs: list | tuple = (), 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) super().__init__(max_workers=max_workers, max_tasks=max_tasks, initializer=initializer, initargs=initargs, context=context)
def shutdown(self, wait=True, *, cancel_futures=False): def shutdown(self, wait=True, *, cancel_futures=False):