improve: better model paths

https://github.com/ltdrdata/ComfyUI-Impact-Subpack/issues/13

__init__.py

@@ -7,7 +7,7 @@
 
 import importlib
 
-version_code = [1, 0]
+version_code = [1, 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})")
 

modules/subpack_nodes.py

@@ -1,12 +1,21 @@
+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)
+
+def update_model_paths(model_path):
+    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'[Impact Subpack] ultralytics_bbox: {os.path.join(model_path, "ultralytics", "bbox")}')
+    logging.info(f'[Impact Subpack] ultralytics_segm: {os.path.join(model_path, "ultralytics", "segm")}')
+
+update_model_paths(folder_paths.models_dir)
+if 'download_model_base' in folder_paths.folder_names_and_paths:
+    update_model_paths(folder_paths.get_folder_paths('download_model_base')[0])
 
 
 class UltralyticsDetectorProvider:

modules/utils.py

@@ -2,6 +2,7 @@ import folder_paths
 from PIL import Image
 import numpy as np
 import cv2
+import torch
 
 
 def add_folder_path_and_extensions(folder_name, full_folder_paths, extensions):

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.1"
 license = { file = "LICENSE.txt" }
 dependencies = ["matplotlib", "ultralytics==8.3.40"]