mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 06:40:48 +08:00
3ddec8ae90
- --panics-when=torch.cuda.OutOfMemory will now correctly panic and exit the worker, giving it time to reply that the execution failed and better dealing with irrecoverable out-of-memory errors - --executor-factory=ProcessPoolExecutor will use a process instead of a thread to execute comfyui workflows when using the worker. When this process panics and exits, it will be correctly replaced, making a more robust worker
16 lines
657 B
Python
16 lines
657 B
Python
# todo: this should be defined in a common place, the fact that the nodes are imported by execution the way that they are is pretty radioactive
|
|
import lazy_object_proxy
|
|
|
|
from comfy.execution_context import current_execution_context
|
|
from comfy.nodes.package import import_all_nodes_in_workspace
|
|
from comfy.nodes.package_typing import ExportedNodes, exported_nodes_view
|
|
|
|
nodes: ExportedNodes = lazy_object_proxy.Proxy(import_all_nodes_in_workspace)
|
|
|
|
|
|
def get_nodes() -> ExportedNodes:
|
|
current_ctx = current_execution_context()
|
|
if len(current_ctx.custom_nodes) == 0:
|
|
return nodes
|
|
return exported_nodes_view(nodes, current_ctx.custom_nodes)
|