298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-16 00:23:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

support pip comfyui-manager

Browse Source
This commit is contained in:
Dr.Lt.Data 2025-03-19 22:24:04 +09:00
parent 3b19fc76e3
commit 6897a1d077
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -10,6 +10,7 @@ from app.logger import setup_logger
import itertools import itertools
import utils.extra_config import utils.extra_config
import logging import logging
import comfyui_manager
if __name__ == "__main__": if __name__ == "__main__":
#NOTE: These do not do anything on core ComfyUI which should already have no communication with the internet, they are for custom nodes. #NOTE: These do not do anything on core ComfyUI which should already have no communication with the internet, they are for custom nodes.
@ -69,6 +70,8 @@ def execute_prestartup_script():
if args.disable_all_custom_nodes: if args.disable_all_custom_nodes:
return return
comfyui_manager.prestartup()
node_paths = folder_paths.get_folder_paths("custom_nodes") node_paths = folder_paths.get_folder_paths("custom_nodes")
for custom_node_path in node_paths: for custom_node_path in node_paths:
possible_modules = os.listdir(custom_node_path) possible_modules = os.listdir(custom_node_path)
@ -261,6 +264,7 @@ def start_comfyui(asyncio_loop=None):
prompt_server = server.PromptServer(asyncio_loop) prompt_server = server.PromptServer(asyncio_loop)
q = execution.PromptQueue(prompt_server) q = execution.PromptQueue(prompt_server)
comfyui_manager.start()
nodes.init_extra_nodes(init_custom_nodes=not args.disable_all_custom_nodes) nodes.init_extra_nodes(init_custom_nodes=not args.disable_all_custom_nodes)
cuda_malloc_warning() cuda_malloc_warning()

1
nodes.py
View File

@ -37,6 +37,7 @@ import importlib
import folder_paths import folder_paths
import latent_preview import latent_preview
import node_helpers import node_helpers
import comfyui_manager
def before_node_execution(): def before_node_execution():
comfy.model_management.throw_exception_if_processing_interrupted() comfy.model_management.throw_exception_if_processing_interrupted()