mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-12 04:27:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding some suggests from the author

Browse Source
This commit is contained in:
Hacker 17082006
2023-02-14 22:20:30 +07:00
parent 721f8510af
commit 493671574e
5 changed files with 188 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
nodes.py
View File

@@ -598,12 +598,19 @@ NODE_CLASS_MAPPINGS = {
"CLIPLoader": CLIPLoader,
}
CUSTOM_NODE_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "custom_nodes")
def load_custom_nodes():
possible_modules = os.listdir("custom_nodes")
possible_modules.remove("put_custom_nodes_here")
possible_modules = os.listdir(CUSTOM_NODE_PATH)
try:
possible_modules.remove("example.py")
possible_modules.remove("example_folder")
except ValueError: pass
for possible_module in possible_modules:
module_path = os.path.join(CUSTOM_NODE_PATH, possible_module)
if os.path.isfile(module_path) and os.path.splitext(module_path)[1] != ".py": continue
try:
custom_nodes = import_module(possible_module, "custom_nodes")
custom_nodes = import_module(possible_module, CUSTOM_NODE_PATH)
if getattr(custom_nodes, "NODE_CLASS_MAPPINGS") is not None:
NODE_CLASS_MAPPINGS.update(custom_nodes.NODE_CLASS_MAPPINGS)
else: