Skip to content

Commit

Permalink
Cleanup.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Sep 10, 2024
1 parent 9fa8faa commit bc94662
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ def execute_script(script_path):
import gc

import logging
from utils import extra_config
import utils.extra_config

if os.name == "nt":
logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
Expand Down Expand Up @@ -201,11 +201,11 @@ def cleanup_temp():

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):
extra_config.load_extra_path_config(extra_model_paths_config_path)
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):
extra_config.load_extra_path_config(config_path)
utils.extra_config.load_extra_path_config(config_path)

nodes.init_extra_nodes(init_custom_nodes=not args.disable_all_custom_nodes)

Expand Down

1 comment on commit bc94662

@rvorias
Copy link

@rvorias rvorias commented on bc94662 Sep 11, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

importing utils.extra_config in main.py breaks my env: "'utils' is not a package"

update: run import utils before doing something like import main

Please sign in to comment.