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

Always use spawn context, even on Linux

Browse Source
This commit is contained in:
Benjamin Berman 2025-07-26 17:42:44 -07:00
parent 2fd5fc81f5
commit 583ddd6b38
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/distributed/process_pool_executor.py
View File

@ -2,6 +2,7 @@ import concurrent.futures
import contextvars
import multiprocessing
import pickle
import logging
from functools import partial
from typing import Callable, Any
@ -9,6 +10,7 @@ from pebble import ProcessPool, ProcessFuture
from ..component_model.executor_types import Executor
logger = logging.getLogger(__name__)
def _wrap_with_context(context_data: bytes, func: Callable, *args, **kwargs) -> Any:
new_ctx: contextvars.Context = pickle.loads(context_data)
@ -22,8 +24,9 @@ class ProcessPoolExecutor(ProcessPool, Executor):
initializer: Callable = None,
initargs: list | tuple = (),
context: multiprocessing.context.BaseContext = None):
if context is None:
context = multiprocessing.get_context('spawn')
if context is not None:
logger.warning(f"A context was passed to a ProcessPoolExecutor when only spawn is supported (context={context})")
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):