From 92aef77a8134321d36f936cf7bc1efaa625cff4d Mon Sep 17 00:00:00 2001 From: atmaranto <35581227+atmaranto@users.noreply.github.com> Date: Wed, 9 Oct 2024 03:43:28 -0500 Subject: [PATCH] Update comfyui_to_python_utils.py --- comfyui_to_python_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/comfyui_to_python_utils.py b/comfyui_to_python_utils.py index 184ff09..6e5e33b 100644 --- a/comfyui_to_python_utils.py +++ b/comfyui_to_python_utils.py @@ -14,7 +14,7 @@ def import_custom_nodes() -> None: """ import asyncio import execution - from nodes import init_custom_nodes + from nodes import init_extra_nodes import server # Creating a new event loop and setting it as the default loop @@ -26,7 +26,7 @@ def import_custom_nodes() -> None: execution.PromptQueue(server_instance) # Initializing custom nodes - init_custom_nodes() + init_extra_nodes(init_custom_nodes=True) def find_path(name: str, path: str = None) -> str: @@ -77,7 +77,7 @@ def add_extra_model_paths() -> None: """ Parse the optional extra_model_paths.yaml file and add the parsed paths to the sys.path. """ - from main import load_extra_path_config + from utils.extra_config import load_extra_path_config extra_model_paths = find_path("extra_model_paths.yaml")