Forgot this in last commit.

This commit is contained in:
comfyanonymous 2024-07-04 21:49:50 -04:00
parent bd2d3e27d7
commit cedbc94cc0
1 changed files with 4 additions and 1 deletions

View File

@ -1890,6 +1890,9 @@ EXTENSION_WEB_DIRS = {}
def load_custom_node(module_path, ignore=set()): def load_custom_node(module_path, ignore=set()):
module_name = os.path.basename(module_path) module_name = os.path.basename(module_path)
if os.path.isfile(module_path):
sp = os.path.splitext(module_path)
module_name = sp[0]
try: try:
logging.debug("Trying to load custom node {}".format(module_path)) logging.debug("Trying to load custom node {}".format(module_path))
if os.path.isfile(module_path): if os.path.isfile(module_path):
@ -2019,7 +2022,7 @@ def init_builtin_extra_nodes():
def init_extra_nodes(init_custom_nodes=True): def init_extra_nodes(init_custom_nodes=True):
import_failed = init_external_custom_nodes() import_failed = init_builtin_extra_nodes()
if init_custom_nodes: if init_custom_nodes:
init_external_custom_nodes() init_external_custom_nodes()