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

Merge ba03485c6d into 4c4be1bba5

Browse Source
This commit is contained in:
hpr 2026-03-14 17:10:15 -05:00 committed by GitHub
commit d15637e4fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nodes.py
View File

@ -14,6 +14,7 @@ import math
import time import time
import random import random
import logging import logging
import psutil
from PIL import Image, ImageOps, ImageSequence from PIL import Image, ImageOps, ImageSequence
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
@ -2309,6 +2310,7 @@ async def init_external_custom_nodes():
base_node_names = set(NODE_CLASS_MAPPINGS.keys()) base_node_names = set(NODE_CLASS_MAPPINGS.keys())
node_paths = folder_paths.get_folder_paths("custom_nodes") node_paths = folder_paths.get_folder_paths("custom_nodes")
node_import_times = [] node_import_times = []
process = psutil.Process()
for custom_node_path in node_paths: for custom_node_path in node_paths:
possible_modules = os.listdir(os.path.realpath(custom_node_path)) possible_modules = os.listdir(os.path.realpath(custom_node_path))
if "__pycache__" in possible_modules: if "__pycache__" in possible_modules:
@ -2329,18 +2331,21 @@ async def init_external_custom_nodes():
logging.info(f"Blocked by policy: {module_path}") logging.info(f"Blocked by policy: {module_path}")
continue continue
rss_before = process.memory_info().rss
time_before = time.perf_counter() time_before = time.perf_counter()
success = await load_custom_node(module_path, base_node_names, module_parent="custom_nodes") success = await load_custom_node(module_path, base_node_names, module_parent="custom_nodes")
node_import_times.append((time.perf_counter() - time_before, module_path, success)) elapsed = time.perf_counter() - time_before
rss_delta = (process.memory_info().rss - rss_before) / (1024 * 1024)
node_import_times.append((elapsed, rss_delta, module_path, success))
if len(node_import_times) > 0: if len(node_import_times) > 0:
logging.info("\nImport times for custom nodes:") logging.info("\nImport times for custom nodes:")
for n in sorted(node_import_times): for n in sorted(node_import_times):
if n[2]: if n[3]:
import_message = "" import_message = ""
else: else:
import_message = " (IMPORT FAILED)" import_message = " (IMPORT FAILED)"
logging.info("{:6.1f} seconds{}: {}".format(n[0], import_message, n[1])) logging.info("{:6.1f} seconds RAM {:>6.1f} MB{}: {}".format(n[0], n[1], import_message, n[2]))
logging.info("") logging.info("")
async def init_builtin_extra_nodes(): async def init_builtin_extra_nodes():