fix: raise exception instead of pass when ultralytics importing is failed

This commit is contained in:
Dr.Lt.Data
2023-11-23 20:18:26 +09:00
parent 57abba847a
commit 5dd0822b83
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -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):
+1 -1
View File
@@ -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)