From 5dd0822b83c579de3ef65082cd376efbb898bbf4 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Thu, 23 Nov 2023 20:18:26 +0900 Subject: [PATCH] fix: raise exception instead of pass when ultralytics importing is failed --- impact/subcore.py | 2 +- impact/subpack_nodes.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/impact/subcore.py b/impact/subcore.py index 0708db7..a9ab4c6 100644 --- a/impact/subcore.py +++ b/impact/subcore.py @@ -12,7 +12,7 @@ try: except Exception as e: print(e) print(f"\n!!!!!\n\n[ComfyUI-Impact-Subpack] If this error occurs, please check the following link:\n\thttps://github.com/ltdrdata/ComfyUI-Impact-Pack/blob/Main/troubleshooting/TROUBLESHOOTING.md\n\n!!!!!\n") - pass + raise e def load_yolo(model_path: str): diff --git a/impact/subpack_nodes.py b/impact/subpack_nodes.py index 51ab73f..13a14e1 100644 --- a/impact/subpack_nodes.py +++ b/impact/subpack_nodes.py @@ -4,7 +4,7 @@ import impact.core as core import impact.subcore as subcore from impact.utils import add_folder_path_and_extensions -print(f"### Loading: ComfyUI-Impact-Pack (Subpack: V0.3.1)") +print(f"### Loading: ComfyUI-Impact-Pack (Subpack: V0.3.2)") model_path = folder_paths.models_dir add_folder_path_and_extensions("ultralytics_bbox", [os.path.join(model_path, "ultralytics", "bbox")], folder_paths.supported_pt_extensions)