From 3d94a74652ebfda0df64a1edbb85849ecc6b2136 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Mon, 16 Dec 2024 00:29:26 +0900
Subject: [PATCH] feat: logging ultralytics model path

---
 __init__.py              | 2 +-
 modules/subpack_nodes.py | 4 ++++
 pyproject.toml           | 2 +-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index e048f1d..030f919 100644
--- a/__init__.py
+++ b/__init__.py
@@ -7,7 +7,7 @@
 
 import importlib
 
-version_code = [1, 0]
+version_code = [1, 0, 1]
 version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 print(f"### Loading: ComfyUI-Impact-Subpack ({version_str})")
 
diff --git a/modules/subpack_nodes.py b/modules/subpack_nodes.py
index ff80287..7546796 100644
--- a/modules/subpack_nodes.py
+++ b/modules/subpack_nodes.py
@@ -1,12 +1,16 @@
+import logging
 import os
 import folder_paths
 from . import subcore
 from . import utils
+import logging
 
 model_path = folder_paths.models_dir
 utils.add_folder_path_and_extensions("ultralytics_bbox", [os.path.join(model_path, "ultralytics", "bbox")], folder_paths.supported_pt_extensions)
 utils.add_folder_path_and_extensions("ultralytics_segm", [os.path.join(model_path, "ultralytics", "segm")], folder_paths.supported_pt_extensions)
 utils.add_folder_path_and_extensions("ultralytics", [os.path.join(model_path, "ultralytics")], folder_paths.supported_pt_extensions)
+logging.info(f'[Inspire Pack] ultralytics_bbox: {os.path.join(model_path, "ultralytics", "bbox")}')
+logging.info(f'[Inspire Pack] ultralytics_segm: {os.path.join(model_path, "ultralytics", "segm")}')
 
 
 class UltralyticsDetectorProvider:
diff --git a/pyproject.toml b/pyproject.toml
index 42f1c83..df5201e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-impact-subpack"
 description = "This node pack provides nodes that complement the Impact Pack, such as the UltralyticsDetectorProvider."
-version = "1.0"
+version = "1.0.1"
 license = { file = "LICENSE.txt" }
 dependencies = ["matplotlib", "ultralytics==8.3.40"]