diff --git a/main.py b/main.py index 40237d7d7..f15725939 100644 --- a/main.py +++ b/main.py @@ -11,14 +11,15 @@ import itertools import utils.extra_config import logging import sys +import importlib.util from comfy_execution.progress import get_progress_state from comfy_execution.utils import get_executing_context from comfy_api import feature_flags if not args.disable_manager: - try: + if importlib.util.find_spec("comfyui_manager"): import comfyui_manager - except Exception: + else: logging.warning(f"\n\nUnable to run comfyui-manager, disabling it. To enable comfyui-manager, run the following command:\ncommand:\n\t{sys.executable} -m pip install --pre comfyui_manager\n") args.disable_manager = True