diff --git a/app/custom_node_manager.py b/app/custom_node_manager.py index 42b0d75ba..59d4f23a9 100644 --- a/app/custom_node_manager.py +++ b/app/custom_node_manager.py @@ -8,7 +8,7 @@ import json import logging from functools import lru_cache -from utils.json_util import merge_json_recursive +from ComfyUI.utils.json_util import merge_json_recursive # Extra locale files to load into main.json diff --git a/main.py b/main.py index ac9d24b7b..baa7bce7e 100644 --- a/main.py +++ b/main.py @@ -8,10 +8,12 @@ import time from comfy.cli_args import args from app.logger import setup_logger import itertools -import utils.extra_config import logging import sys +sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +import ComfyUI.utils.extra_config + 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. os.environ['HF_HUB_DISABLE_TELEMETRY'] = '1' @@ -24,11 +26,11 @@ def apply_custom_paths(): # extra model paths extra_model_paths_config_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "extra_model_paths.yaml") if os.path.isfile(extra_model_paths_config_path): - utils.extra_config.load_extra_path_config(extra_model_paths_config_path) + ComfyUI.utils.extra_config.load_extra_path_config(extra_model_paths_config_path) if args.extra_model_paths_config: for config_path in itertools.chain(*args.extra_model_paths_config): - utils.extra_config.load_extra_path_config(config_path) + ComfyUI.utils.extra_config.load_extra_path_config(config_path) # --output-directory, --input-directory, --user-directory if args.output_directory: