mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-11 02:52:36 +08:00
7d512fa9c3
Adds the isolation system foundation: ComfyUIAdapter, extension loader, manifest discovery, child/host process hooks, RPC bridge, runtime helpers, SHM forensics, and the --use-process-isolation CLI flag. pyisolate added to requirements.txt. .pyisolate_venvs/ added to .gitignore.
127 lines
4.0 KiB
Python
127 lines
4.0 KiB
Python
# pylint: disable=import-outside-toplevel,logging-fstring-interpolation
|
|
# Child process initialization for PyIsolate
|
|
import logging
|
|
import os
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def is_child_process() -> bool:
|
|
return os.environ.get("PYISOLATE_CHILD") == "1"
|
|
|
|
|
|
def _load_extra_model_paths() -> None:
|
|
"""Load extra_model_paths.yaml so the child's folder_paths has the same search paths as the host.
|
|
|
|
The host loads this in main.py:143-145. The child is spawned by
|
|
pyisolate's uds_client.py and never runs main.py, so folder_paths
|
|
only has the base model directories. Any isolated node calling
|
|
folder_paths.get_filename_list() in define_schema() would get empty
|
|
results for folders whose files live in extra_model_paths locations.
|
|
"""
|
|
import folder_paths # noqa: F401 — side-effect import; load_extra_path_config writes to folder_paths internals
|
|
from utils.extra_config import load_extra_path_config
|
|
|
|
extra_config_path = os.path.join(
|
|
os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))),
|
|
"extra_model_paths.yaml",
|
|
)
|
|
if os.path.isfile(extra_config_path):
|
|
load_extra_path_config(extra_config_path)
|
|
|
|
|
|
def initialize_child_process() -> None:
|
|
logger.warning("][ DIAG:child_hooks initialize_child_process START")
|
|
if os.environ.get("PYISOLATE_IMPORT_TORCH", "1") != "0":
|
|
_load_extra_model_paths()
|
|
_setup_child_loop_bridge()
|
|
|
|
# Manual RPC injection
|
|
try:
|
|
from pyisolate._internal.rpc_protocol import get_child_rpc_instance
|
|
|
|
rpc = get_child_rpc_instance()
|
|
logger.warning("][ DIAG:child_hooks RPC instance: %s", rpc is not None)
|
|
if rpc:
|
|
_setup_proxy_callers(rpc)
|
|
logger.warning("][ DIAG:child_hooks proxy callers configured with RPC")
|
|
else:
|
|
logger.warning("][ DIAG:child_hooks NO RPC — proxy callers cleared")
|
|
_setup_proxy_callers()
|
|
except Exception as e:
|
|
logger.error(f"][ DIAG:child_hooks Manual RPC Injection failed: {e}")
|
|
_setup_proxy_callers()
|
|
|
|
_setup_logging()
|
|
|
|
|
|
def _setup_child_loop_bridge() -> None:
|
|
import asyncio
|
|
|
|
main_loop = None
|
|
try:
|
|
main_loop = asyncio.get_running_loop()
|
|
except RuntimeError:
|
|
try:
|
|
main_loop = asyncio.get_event_loop()
|
|
except RuntimeError:
|
|
pass
|
|
|
|
if main_loop is None:
|
|
return
|
|
|
|
try:
|
|
from .proxies.base import set_global_loop
|
|
|
|
set_global_loop(main_loop)
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def _setup_prompt_server_stub(rpc=None) -> None:
|
|
try:
|
|
from .proxies.prompt_server_impl import PromptServerStub
|
|
|
|
if rpc:
|
|
PromptServerStub.set_rpc(rpc)
|
|
elif hasattr(PromptServerStub, "clear_rpc"):
|
|
PromptServerStub.clear_rpc()
|
|
else:
|
|
PromptServerStub._rpc = None # type: ignore[attr-defined]
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to setup PromptServerStub: {e}")
|
|
|
|
|
|
def _setup_proxy_callers(rpc=None) -> None:
|
|
try:
|
|
from .proxies.folder_paths_proxy import FolderPathsProxy
|
|
from .proxies.helper_proxies import HelperProxiesService
|
|
from .proxies.model_management_proxy import ModelManagementProxy
|
|
from .proxies.progress_proxy import ProgressProxy
|
|
from .proxies.prompt_server_impl import PromptServerStub
|
|
from .proxies.utils_proxy import UtilsProxy
|
|
|
|
if rpc is None:
|
|
FolderPathsProxy.clear_rpc()
|
|
HelperProxiesService.clear_rpc()
|
|
ModelManagementProxy.clear_rpc()
|
|
ProgressProxy.clear_rpc()
|
|
PromptServerStub.clear_rpc()
|
|
UtilsProxy.clear_rpc()
|
|
return
|
|
|
|
FolderPathsProxy.set_rpc(rpc)
|
|
HelperProxiesService.set_rpc(rpc)
|
|
ModelManagementProxy.set_rpc(rpc)
|
|
ProgressProxy.set_rpc(rpc)
|
|
PromptServerStub.set_rpc(rpc)
|
|
UtilsProxy.set_rpc(rpc)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to setup child singleton proxy callers: {e}")
|
|
|
|
|
|
def _setup_logging() -> None:
|
|
logging.getLogger().setLevel(logging.INFO)
|