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

Improve vanilla node importing for execution nodes

Browse Source
This commit is contained in:
doctorpangloss 2024-10-04 10:56:43 -07:00
parent 144fe6c421
commit de45dd50c5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/nodes/vanilla_node_importing.py
View File

@ -162,11 +162,15 @@ def mitigated_import_of_vanilla_custom_nodes() -> ExportedNodes:
# found. we're in the middle of executing the import of execution and server, in all likelihood, so like all things, # found. we're in the middle of executing the import of execution and server, in all likelihood, so like all things,
# the way community custom nodes is pretty radioactive # the way community custom nodes is pretty radioactive
from ..cmd import cuda_malloc, folder_paths, latent_preview from ..cmd import cuda_malloc, folder_paths, latent_preview
from .. import graph, graph_utils, caching
from .. import node_helpers from .. import node_helpers
for module in (cuda_malloc, folder_paths, latent_preview, node_helpers): for module in (cuda_malloc, folder_paths, latent_preview, node_helpers):
module_short_name = module.__name__.split(".")[-1] module_short_name = module.__name__.split(".")[-1]
sys.modules[module_short_name] = module sys.modules[module_short_name] = module
sys.modules['nodes'] = base_nodes sys.modules['nodes'] = base_nodes
sys.modules['comfy_execution.graph'] = graph
sys.modules['comfy_execution.graph_utils'] = graph_utils
sys.modules['comfy_execution.caching'] = caching
from ..cmd import execution, server from ..cmd import execution, server
for module in (execution, server): for module in (execution, server):
module_short_name = module.__name__.split(".")[-1] module_short_name = module.__name__.split(".")[-1]